Fixed broken testcase