updated the test case
bug fix