Small fix in the ut
Merge pull request #125 from jaytaph/wip/utfix

Small fix in the ut