Fix test case failed issue