Quick Fix for Upgrade Function