Fix broken testcase