Fix testcase