diff --git a/setup.py b/setup.py
index 694844d9e374f955a19c89f2ee8fd1075dee4bb5..6e7664a45f330597f4985fc894839c18fa362d3e 100644
--- a/setup.py
+++ b/setup.py
@@ -180,7 +180,7 @@ class CustomBuildExt(build_ext):
         logger.info(f"Copied {so_name} to {dst_so_path}")
 
         # Copy generated Python files to Python package directory
-        auto_generate_dir = os.path.join(build_extension_dir, "auto_generate")
+        auto_generate_dir = os.path.join(build_extension_dir, ext_name + "_auto_generate")
         if os.path.exists(auto_generate_dir):
             generated_files = ["gen_ops_def.py", "gen_ops_prim.py"]
             for gen_file in generated_files: