Fix bug with check screen.