[UPDATE] Check for upgrade later