diff --git a/setup.py b/setup.py
index e0eaf4f5262..f1cc3119fe7 100644
--- a/setup.py
+++ b/setup.py
@@ -249,6 +249,7 @@ def _generate_python_protos(self, path: str):
                 self.python_folder,
             ]
             + proto_files,
+            env=os.environ
         )
 
     def run(self):
@@ -334,6 +335,8 @@ def _generate_go_protos(self, path: str):
         proto_files = glob.glob(os.path.join(self.proto_folder, path))
 
         try:
+            e = os.environ.copy()
+            e["PATH"] = self.path_val
             subprocess.check_call(
                 self.go_protoc
                 + [
@@ -347,7 +350,7 @@ def _generate_go_protos(self, path: str):
                     "--go-grpc_opt=module=github.com/feast-dev/feast/go/protos",
                 ]
                 + proto_files,
-                env={"PATH": self.path_val},
+                env=e,
             )
         except CalledProcessError as e:
             print(f"Stderr: {e.stderr}")