Fix linear regression