Fix for neug_upgrade.py.