fix a bug in test case.