diff --git a/setup.py b/setup.py
index bad433ce..60fe6295 100644
--- a/setup.py
+++ b/setup.py
@@ -1,3 +1,4 @@
+import numpy
 from pprint import pprint
 from urllib.request import urlretrieve
 import sys
@@ -44,7 +45,7 @@ def build_extension(self, ext):
 ext = [Extension(name='.'.join(['ivfhnsw', '_' + name]),
                  sources=[os.path.join('interface', '.'.join([name, 'i']))],
                  swig_opts=['-Iinclude', '-c++'],
-                 include_dirs=['include', os.curdir],
+                 include_dirs=['include', os.curdir, numpy.get_include()],
                  libraries=['ivfhnsw', 'hnswlib', 'faiss', 'gomp', 'lapack',],
                  extra_compile_args=['-std=c++11', '-static'],)
                  for name in names]
@@ -56,7 +57,10 @@ def build_extension(self, ext):
     ext_modules=ext,
     package_dir={'': python_src},
     packages=find_packages(python_src),
-    setup_requires=['pytest-runner'],
+    setup_requires=[
+        'pytest-runner',
+        'numpy',
+    ],
     install_requires=[
         'numpy',
     ],