Fixed bug with 2D plots