diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index 1edd74859e..569f8c3904 100644
--- a/sdk/python/setup.py
+++ b/sdk/python/setup.py
@@ -324,7 +324,7 @@ def run(self):
             self._generate_go_protos(f"feast/{sub_folder}/*.proto")
 
 
-class BuildGoEmbeddedCommand(build_py):
+class BuildGoEmbeddedCommand(Command):
     description = "Builds Go embedded library"
     user_options = []