minor bug fixed for testcase