public function getPaymentStateBadge($stateOrPayment)
    {
        $state = $stateOrPayment instanceof PaymentInterface ? $stateOrPayment->getState() : $stateOrPayment;

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