Fix py_modules on setup.py