swap argument order for STATE

since all the other refinement invariants take the context first