/** * @inheritdoc */ public function removeState(StateInterface $state) { if ($this->hasState($state)) { $state->setCountry(null); $this->states->removeElement($state); } return $this; }