Fixed the plot issue