Fixing tests

I broke these tests ages ago! Oops.