small fix of unit-test