fix test error.
Merge pull request #167 from forumbyr/master

fix test error.