Update main.py

fixed a variable name