fixed error in main.py