Replace check_screen method with assert_screen.