Fixing py_modules