Adds missing state arg to draw_widget.