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

Fix tests again