Fix main script in setup.py/freeze.py