Fix testcase (hopefully)