Fixed reference to main in setup.py