fixes broken tests
Merge pull request #16 from penzance/feature/tests

fixes broken tests