Fixed upgrade test