Correct typo in main.py