Made a minor fix for testcase