Correct typo.
Restore saving via matplotlib.