diff --git a/setup.py b/setup.py
index 77097e18176..d4db43068e4 100644
--- a/setup.py
+++ b/setup.py
@@ -111,7 +111,9 @@ def get_build_version():
     if "tf" in sys.argv:
         sys.argv.remove("tf")
         cfg_key = "neural_compressor_tf"
-
+    
+    cfg_key = "neural_compressor_pt"
+    
     ext_modules = []
     cmdclass = {}
     project_name = PKG_INSTALL_CFG[cfg_key].get("project_name")