Fix failing testcase