fixed up the tests i broke