Ensure wait_for_press works for subsequent calls