diff --git a/setup.py b/setup.py
index 1e99d5e..805acea 100644
--- a/setup.py
+++ b/setup.py
@@ -20,6 +20,7 @@
 
 setup(
     name="simple_knn",
+    setup_requires=['torch'],
     ext_modules=[
         CUDAExtension(
             name="simple_knn._C",
@@ -28,9 +29,9 @@
             "simple_knn.cu",
             "ext.cpp"],
             extra_compile_args={"nvcc": [], "cxx": cxx_compiler_flags})
-        ],
+    ],
     cmdclass={
         'build_ext': BuildExtension
     },
-    version='1.0.0'
+    version='1.0.0',
 )