Fix redrawing