fix test
updates
Merge pull request #33 from stevelacy/master

Fix tests