Fixing the screen update