Add implicit context passing through State.