fix py modules in setup