Update include order

Must include `FailureState` before referencing it.