Fix test, now the output is correct