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

fix unit test problem.