Fix Bats tests.
Merge pull request #8 from duggan/bats-tests

Fix Bats tests