add a missing plt.close('all')