bug fix for the test case