Fix a typo in main.py