fix tests, yay