Upgrade with controller