fix unit test that i broke