Widget bug fixes