Updated main.py

Changed a small error.