Fix for widget update bug