FIX: Use the currently upgrading key to check the version