fix tests
Merge pull request #180 from devoncarew/devoncarew_fix_tests

fix tests