Fixed error with widget disappearing.