More flattening IRState.