diff --git a/scripts/Python/python-wrapper.swig b/scripts/Python/python-wrapper.swig
index 4723a9c..55a5910 100644
--- a/scripts/Python/python-wrapper.swig
+++ b/scripts/Python/python-wrapper.swig
@@ -45,7 +45,13 @@
             return NULL;
     }
 
-    if (!PyDict_Check(pmodule))
+    if (PyType_Check(pmodule))
+    {
+        main_dict = ((PyTypeObject*)pmodule)->tp_dict;
+        if (!main_dict)
+            return NULL;
+    }
+    else if (!PyDict_Check(pmodule))
     {
         main_dict = PyModule_GetDict (pmodule);
         if (!main_dict)