working on widget. bug fix