diff --git a/src/wrap_cl.cpp b/src/wrap_cl.cpp
index 50a482016ebda749ec4031f9d7cb92ab200dfa12..b9393ed0f4963968d7006d686e72c349eebc20ad 100644
--- a/src/wrap_cl.cpp
+++ b/src/wrap_cl.cpp
@@ -43,7 +43,11 @@ extern void pyopencl_expose_mempool(py::module &m);
 
 static bool import_numpy_helper()
 {
-  import_array1(false);
+#ifdef PYPY_VERSION
+      import_array();
+#else
+      import_array1(false);
+#endif
   return true;
 }