diff --git a/src/Spryker/Zed/MerchantDataImport/Business/Address/Step/MerchantKeyToIdMerchantStep.php b/src/Spryker/Zed/MerchantDataImport/Business/Address/Step/MerchantKeyToIdMerchantStep.php
index <HASH>..<HASH> 100644
--- a/src/Spryker/Zed/MerchantDataImport/Business/Address/Step/MerchantKeyToIdMerchantStep.php
+++ b/src/Spryker/Zed/MerchantDataImport/Business/Address/Step/MerchantKeyToIdMerchantStep.php
@@ -56,12 +56,13 @@ class MerchantKeyToIdMerchantStep implements DataImportStepInterface
         /** @var \Orm\Zed\Merchant\Persistence\SpyMerchantQuery $merchantQuery */
         $merchantQuery = SpyMerchantQuery::create()
             ->select(SpyMerchantTableMap::COL_ID_MERCHANT);
-        $merchantEntity = $merchantQuery->findOneByMerchantKey($merchantKey);
+        /** @var int $idMerchant */
+        $idMerchant = $merchantQuery->findOneByMerchantKey($merchantKey);
 
-        if (!$merchantEntity) {
+        if (!$idMerchant) {
             throw new EntityNotFoundException(sprintf('Could not find Merchant by key "%s"', $merchantKey));
         }
 
-        return $merchantEntity->getIdMerchant();
+        return $idMerchant;
     }
 }
Updated getIdMerchant functionality