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

fix test unit problem.