Fix memory testing linear regression