Fix output not being displayed properly.