diff --git a/src/Observers/EeStoreWebsiteValidatorObserver.php b/src/Observers/EeStoreWebsiteValidatorObserver.php
index 637bd09..4fecbe7 100644
--- a/src/Observers/EeStoreWebsiteValidatorObserver.php
+++ b/src/Observers/EeStoreWebsiteValidatorObserver.php
@@ -1,7 +1,7 @@
 <?php
 
 /**
- * TechDivision\Import\Callbacks\ArrayValidatorCallback
+ * TechDivision\Import\Product\Ee\Observers\StoreWebsiteValidatorObserver
  *
  * PHP version 7
  *
@@ -19,7 +19,7 @@
 use TechDivision\Import\Product\Utils\MemberNames;
 
 /**
- * storeview validator callback implementation.
+ * Store view validator implementation.
  *
  * @author    MET <met@techdivision.com>
  * @copyright 2024 TechDivision GmbH <info@techdivision.com>
@@ -32,10 +32,10 @@ class EeStoreWebsiteValidatorObserver extends StoreWebsiteValidatorObserver
     /**
      * @return void
      */
-    public function setLastEntityRowId(): void
+    public function setLastEntityRowId($sku): void
     {
-        if (!$this->hasBeenProcessed($this->getValue(ColumnKeys::SKU))) {
-            $this->entity = $this->loadProduct($this->getValue(MemberNames::SKU));
+        if (!$this->hasBeenProcessed($sku)) {
+            $this->entity = $this->loadProduct($sku);
             $this->setLastEntityId($this->entity[MemberNames::ROW_ID]);
             $this->lastEntityId = $this->getLastEntityId();
         }