Update TestRunStateTests to reflect previous changes.