FIx test case failed