bug fix
Test case updated