Fix to unit test.