Update main.py

Temporary fix for #147