fix build

1 files changed, 2 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 5958dd8..da5894c 100644
--- a/setup.py
+++ b/setup.py
@@ -1,13 +1,14 @@
 from distutils.core import setup
 from distutils.extension import Extension
 from Cython.Build import cythonize
+import numpy
 
 all_extensions = Extension("*", ["pyisda/*.pyx"],
                            include_dirs = ['c_layer'],
                            libraries = ["cds"])
 
 c_extension = Extension("pyisda.flat_hazard",
-                        include_dirs = ['c_layer'],
+                        include_dirs = ['c_layer', numpy.get_include()],
                         sources = ['pyisda/flat_hazard.pyx', 'c_layer/cdsbootstrap.c'],
                         libraries = ['cds'])
 all_extensions = cythonize([c_extension, all_extensions], nthreads = 4,