Fix Widget unit tests