fix(tests): fix failing tests that @caitp broke

Oops, my bad!

Closes #37