main.py fix