fixed a unit test error.