public function getOrderStateBadge($stateOrOrder)
    {
        $state = $stateOrOrder instanceof OrderInterface ? $stateOrOrder->getState() : $stateOrOrder;

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