Fix broken plot_util (Previously called plt.gca() before generating a plot)