/**
  * {@inheritDoc}
  */
 public function save(State $state, $sagaId)
 {
     if ($state->isDone()) {
         unset($this->states[$sagaId][$state->getId()]);
     } else {
         $this->states[$sagaId][$state->getId()] = $state;
     }
 }