[RefundType] Change comparison of refund type

Sylius · Jul 6, 2021 · e5e9b15 · e5e9b15
1 parent 38d75ff
commit e5e9b15
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/Provider/RemainingTotalProvider.php b/src/Provider/RemainingTotalProvider.php
@@ -18,7 +18,6 @@
 use Sylius\Component\Order\Model\AdjustableInterface;
 use Sylius\Component\Resource\Repository\RepositoryInterface;
 use Sylius\RefundPlugin\Entity\RefundInterface;
-use Sylius\RefundPlugin\Model\RefundType;
 use Sylius\RefundPlugin\Model\RefundTypeInterface;
 use Webmozart\Assert\Assert;
 
@@ -63,7 +62,7 @@ public function getTotalLeftToRefund(int $id, RefundTypeInterface $type): int
 
     private function getRefundUnitTotal(int $id, RefundTypeInterface $refundType): int
     {
-        if ($refundType->equals(RefundType::orderItemUnit())) {
+        if ($refundType->getValue() === RefundTypeInterface::ORDER_ITEM_UNIT) {
             /** @var OrderItemUnitInterface $orderItemUnit */
             $orderItemUnit = $this->orderItemUnitRepository->find($id);
             Assert::notNull($orderItemUnit);