Pass in the stateful object when fetching the StateActionPair