Fix missing widget

Sorry, was blind...