diff --git a/unwrap2D/setup.py b/unwrap2D/setup.py
index cc780a48..848d43b5 100644
--- a/unwrap2D/setup.py
+++ b/unwrap2D/setup.py
@@ -28,4 +28,5 @@ setup(
     #                        ),
     cmdclass = {'build_ext': build_ext},
     ext_modules = ext_modules,
+    py_modules = ['unwrap',]
     )