diff --git a/model/utils/nms/build.py b/model/utils/nms/build.py
index ddd1787bf9e2ea17303454b141bfa88d676d1091..be916aad5b47ff32bcc583095d33c38de45e5f1f 100644
--- a/model/utils/nms/build.py
+++ b/model/utils/nms/build.py
@@ -1,10 +1,12 @@
 from distutils.core import setup
 from distutils.extension import Extension
 from Cython.Distutils import build_ext
-
-ext_modules = [Extension("_nms_gpu_post", ["_nms_gpu_post.pyx"])]
+import numpy
+#ext_modules = [Extension("_nms_gpu_post", ["_nms_gpu_post.pyx"])]
+ext_modules = [Extension("_nms_gpu_post", ["_nms_gpu_post.pyx"],
+ include_dirs=[numpy.get_include()])]
 setup(
-    name="Hello pyx",
+    name="nms pyx",
     cmdclass={'build_ext': build_ext},
     ext_modules=ext_modules
 )