fix bug
Merge pull request #157 from forumbyr/master

fix bug