fixing the damn unit test that was failing