diff --git a/src/EntityStagingImport.php b/src/EntityStagingImport.php
old mode 100644
new mode 100755
index 2778702..37b243e
--- a/src/EntityStagingImport.php
+++ b/src/EntityStagingImport.php
@@ -116,7 +116,6 @@ class EntityStagingImport {
       if ($founded_key) {
         unset($process['process_definition'][$founded_key]);
       }
-
       $config = [
         'id' => $migration_id . '_' . $language,
         'migration_tags' => ['entity_staging'],
@@ -128,7 +127,7 @@ class EntityStagingImport {
         'migration_group' => 'entity_staging',
         'source' => [
           'plugin' => 'entity_staging_json',
-          'input_path' => '../staging/' . $entity_type_id . '/' . $language . '/' . $bundle_id . '.json',
+          'input_path' => $export_path . '/' . $entity_type_id . '/' . $language . '/' . $bundle_id . '.json',
         ],
         'process' => $process['process_definition'],
         'destination' => [
diff --git a/src/Plugin/migrate/source/EntityStagingJson.php b/src/Plugin/migrate/source/EntityStagingJson.php
index caa941f..de96324 100755
--- a/src/Plugin/migrate/source/EntityStagingJson.php
+++ b/src/Plugin/migrate/source/EntityStagingJson.php
@@ -23,11 +23,21 @@ class EntityStagingJson extends SourcePluginBase {
   protected $input_path;
 
   /**
+   *  Get staging directory
+   *
+   * @var string
+   */
+  protected $staging_directory;
+
+  /**
    * {@inheritdoc}
    */
   public function __construct(array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration, array $namespaces = []) {
     parent::__construct($configuration, $plugin_id, $plugin_definition, $migration);
     $this->input_path = $configuration['input_path'];
+    $config = \Drupal::config('entity_staging.settings');
+    $this->staging_directory = $config->get('staging_directory');
+
     $this->iterator = NULL;
   }
 
@@ -133,7 +143,7 @@ class EntityStagingJson extends SourcePluginBase {
       }
 
       if ($key == 'uri') {
-        $row->setSourceProperty('filepath', realpath('../staging/files') . '/' . str_replace('public://', '', $value));
+        $row->setSourceProperty('filepath', realpath($this->staging_directory . '/files') . '/' . str_replace('public://', '', $value));
       }
 
       if (empty($item)) {