fixing things for 2D plots