CLN: Regression plots fixes

Closes #890.