diff --git a/setup.py b/setup.py
index 1de29990..b979783e 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,9 @@
 
 from pynpm import NPMPackage
 from setuptools import Command, setup
+from setuptools.command.develop import develop
 from setuptools.command.egg_info import egg_info
+from setuptools.command.install import install
 
 ROOT = Path(__file__).parent
 sys.path.append(str(ROOT))
@@ -29,7 +31,7 @@ class DecoratedCommand(command):
 
         def run(self):
             """Run the command"""
-            if sys.argv[1] == "sdist":
+            if sys.argv[1] == "sdist" or sys.argv[1] == "dist_info":
                 from generate_source import generate_source
 
                 npm = NPMPackage(ROOT / "js" / "package.json")
@@ -62,4 +64,10 @@ def get_data_files():
     ]
 
 
-setup(cmdclass={"egg_info": js_prerelease(egg_info)})
+setup(
+    cmdclass={
+        "egg_info": js_prerelease(egg_info),
+        "install": js_prerelease(install),
+        "develop": js_prerelease(develop),
+    }
+)