Load upgrades properly

Fixes a small bug where upgrades were not loaded properly.