Minor fix to make more flexible with new widget
Merge pull request #52 from ordrin/dev_polina

Minor fix to make more flexible with new widget