Fix py_modules syntax in setup.py