fix the tests
Merge pull request #72 from uwej711/fix_tests

fix the tests