bcf564f
diff --git a/setup.py b/setup.py
bcf564f
index 233492d..2dcc7eb 100644
bcf564f
--- a/setup.py
bcf564f
+++ b/setup.py
bcf564f
@@ -1,6 +1,10 @@
bcf564f
-from numpy.distutils.core import setup, Extension
bcf564f
-from numpy.distutils.misc_util import get_numpy_include_dirs 
bcf564f
-import os, sys
bcf564f
+from distutils.core import setup
bcf564f
+from distutils.extension import Extension
bcf564f
+import os, sys, numpy
bcf564f
+
bcf564f
+def get_numpy_include_dirs():
bcf564f
+    include_dirs = [ numpy.get_include() ]
bcf564f
+    return include_dirs
bcf564f
 
bcf564f
 mods = ['gradunwarp.core.coeffs', 'gradunwarp.core.globals', 
bcf564f
         'gradunwarp.core.__init__', 'gradunwarp.__init__',