diff --git a/ViewModel/Warranty.php b/ViewModel/Warranty.php
index 9e0866aa..1d629f03 100644
--- a/ViewModel/Warranty.php
+++ b/ViewModel/Warranty.php
@@ -413,10 +413,10 @@ public function itemHasLeadWarrantyInQuote($orderItem): bool
      * Check does quote have warranty item for the item
      * Kept for backwards compatibility with Hyva module
      *
-     * @param string $sku
+     * @param int $quoteItemId
      * @return bool
      */
-    public function isWarrantyInQuote(string $sku): bool
+    public function isWarrantyInQuote(int $quoteItemId): bool
     {
         try {
             $quote = $this->checkoutSession->getQuote();
@@ -424,7 +424,7 @@ public function isWarrantyInQuote(string $sku): bool
             $quote = null;
         }
         if ($quote) {
-            $hasWarranty = $this->hasWarranty($quote, $sku);
+            $hasWarranty = $this->hasWarranty($quote, $quoteItemId);
         }
         return $hasWarranty ?? false;
     }