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

fix unit test problem