update AutoRegression to work with affine