fixed main.py small error