fix testcases 2 and 3