diff --git a/catnip_uploader/catnip_uploader.py b/catnip_uploader/catnip_uploader.py
index 7d6de1f..fa7473c 100755
--- a/catnip_uploader/catnip_uploader.py
+++ b/catnip_uploader/catnip_uploader.py
@@ -427,5 +427,6 @@ def main():
     catnip_uploader = CatnipUploader()
     catnip_uploader.app()
 
+
 if __name__ == "__main__":
     main()
diff --git a/catnip_uploader/setup.py b/catnip_uploader/setup.py
index 7b65a48..73e528b 100644
--- a/catnip_uploader/setup.py
+++ b/catnip_uploader/setup.py
@@ -7,10 +7,10 @@
     author="Electronic Cats",
     url="https://github.com/ElectronicCats/CatSniffer-Tools",
     packages=find_packages(include=["modules", "modules.*"]),
-    py_modules=["catnip"],
+    py_modules=["catnip_uploader", "cc2538"],
     entry_points={
         "console_scripts": [
-            "catnip=catnip:main",
+            "catnip=catnip_uploader:main",
         ],
     },
     install_requires=[