Fixing tests

Derp