Removed unused import within upgrade.py