small fix on unit test