Fixes issue #75 where TbHeroUnit is not working propery when used via beginWidget endWidget

Fixes issue #75 where TbHeroUnit is not working propery when used via beginWidget endWidget