Correct widget code