Fix how the screen is updated