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