fixed issue in main.py