Oh yeah, forgot to fix the tests.