fix tests i broke