fixing tests
Merge pull request #3 from bradpsheehan/fixing_tests

fixing tests