fix tests that i broke