Add missing redraw