Update the output for 2D.