Update main.py

I changed the same line