bug fix at test case