Make reading state work for non-AR models?