Fix testcases