diff --git a/setup.py b/setup.py
index 0bfecaff2a3d5474b06e4f34a8c1718283b533d7..427ded2549a66c22273e48ac75a0d552b38d4dd2 100644
--- a/setup.py
+++ b/setup.py
@@ -58,7 +58,7 @@ def main():
     from aksetup_helper import (hack_distutils, get_config, setup,
             NumpyExtension, 
             check_git_submodules)
-
+    from setuptools import Extension
     check_git_submodules()
 
     hack_distutils()
@@ -211,7 +211,7 @@ def main():
           
             ext_package="pyopencl",
             ext_modules=[
-                NumpyExtension("_wrapcl",
+                Extension("_wrapcl",
                                ["src/c_wrapper/wrap_cl.cpp",
                                 "src/c_wrapper/wrap_constants.cpp",
                                 #"src/c_wrapper/wrap_mempool.cpp",