Correct bug which removed some output