Fixed bug with widget