Fix testcase failing because of change of output