Fix screen redrawing