fix unit test problem
Merge pull request #203 from forumbyr/master

fix unit test problem