Fix a few plotting issues.

Possible fix for #269