Fix bug with writing out the distribute_setup.py