fix linear regression tests