fix other bug in unit test