fixed a unit test