fix tests
Merge pull request #1 from spheric/fix_tests

fix tests