diff --git a/setup.py b/setup.py
index 54f88db..f872b8e 100644
--- a/setup.py
+++ b/setup.py
@@ -344,10 +344,10 @@ setup(
     install_requires=[
         "torch",
         "einops",
-        "packaging",
-        "ninja",
     ],
     setup_requires=[
-        "psutil"
+        "packaging",
+        "psutil",
+        "ninja",
     ],
 )