diff --git a/app/code/Magento/ImportExport/Model/Import.php b/app/code/Magento/ImportExport/Model/Import.php
index 896a0bc3237aa..07f1913169e11 100644
--- a/app/code/Magento/ImportExport/Model/Import.php
+++ b/app/code/Magento/ImportExport/Model/Import.php
@@ -523,8 +523,6 @@ private function importSourceCallback()
     {
         $ids = $this->getImportIds();
         $this->_getEntityAdapter()->setIds($ids);
-        $this->setData('entity', $this->getDataSourceModel()->getEntityTypeCode($ids));
-        $this->setData('behavior', $this->getDataSourceModel()->getBehavior($ids));
 
         //Validating images temporary directory path if the constraint has been provided
         if ($this->hasData('images_base_directory')
diff --git a/app/code/Magento/ImportExport/Test/Unit/Model/ImportTest.php b/app/code/Magento/ImportExport/Test/Unit/Model/ImportTest.php
index f112b051035ca..4f5b477eb9581 100644
--- a/app/code/Magento/ImportExport/Test/Unit/Model/ImportTest.php
+++ b/app/code/Magento/ImportExport/Test/Unit/Model/ImportTest.php
@@ -320,9 +320,6 @@ public function testImportSource()
             ->method('getEntityTypeCode')
             ->willReturn($entityTypeCode);
         $behaviour = 'behaviour';
-        $this->_importData->expects($this->once())
-            ->method('getBehavior')
-            ->willReturn($behaviour);
         $this->import->expects($this->any())
             ->method('getDataSourceModel')
             ->willReturn($this->_importData);
@@ -389,9 +386,6 @@ public function testImportSourceException()
             ->method('getEntityTypeCode')
             ->willReturn($entityTypeCode);
         $behaviour = 'behaviour';
-        $this->_importData->expects($this->any())
-            ->method('getBehavior')
-            ->willReturn($behaviour);
         $this->import->expects($this->any())
             ->method('getDataSourceModel')
             ->willReturn($this->_importData);