small fix for unit test.