[FIX]Fixed traceback in test case.