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();
     }