Remove unnecessary plt.gca() calls.