From 215f60a71449aeadde7c7d3c34be15e6ab13aa00 Mon Sep 17 00:00:00 2001
From: Nicolas MELONI <nicolas.meloni@synolia.com>
Date: Thu, 28 Sep 2023 15:43:40 +0200
Subject: [PATCH] SYL-216 - fix refund decorator for shipment

---
 src/Creator/RefundUnitsCommandCreatorDecorator.php | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Creator/RefundUnitsCommandCreatorDecorator.php b/src/Creator/RefundUnitsCommandCreatorDecorator.php
index 00a0578e..8e5f0cc9 100644
--- a/src/Creator/RefundUnitsCommandCreatorDecorator.php
+++ b/src/Creator/RefundUnitsCommandCreatorDecorator.php
@@ -21,6 +21,7 @@
 use Sylius\RefundPlugin\Model\OrderItemUnitRefund;
 use Sylius\RefundPlugin\Model\RefundType;
 use Sylius\RefundPlugin\Model\ShipmentRefund;
+use Sylius\RefundPlugin\Model\UnitRefundInterface;
 use Symfony\Component\HttpFoundation\Request;
 use Symfony\Contracts\Translation\TranslatorInterface;
 use Webmozart\Assert\Assert;
@@ -131,7 +132,7 @@ private function getTotalRefundAmount(array $units): int
         return $total;
     }
 
-    private function getAmount(OrderItemUnitRefund $unit): int
+    private function getAmount(UnitRefundInterface $unit): int
     {
         return $unit->total();
     }