Following the rules for contrib
Fixing y axis to take predictor and baseline into account