public function getShipmentStateBadge($stateOrShipment)
    {
        $state = $stateOrShipment instanceof ShipmentInterface ? $stateOrShipment->getState() : $stateOrShipment;

        return sprintf(
            '<span class="label label-%s">%s</span>',
            ShipmentStates::getTheme($state),
            $this->getShipmentStateLabel($state)
        );
    }