fix the tests, yay