Fix tests
Merge pull request #6 from wilsonge/patch-26

Fix tests