fixing the widget bug