fixing the tests
Merge pull request #116 from Blooie/master

fixing the tests