Assertion fix for linear regression response dimension