Didn't realize I broke these tests. Here's a fix! Sorry!
Merge pull request #35 from sourishkrout/master

Cleaning up after myself