Continue writing upgrade test class