[Story20][xuehai] - fixed a bug for the unit test