Update main.py

remove suffix checking
fix for previous patch