Fix bad unit test