Same bug fix for linear regression