diff --git a/dvc.lock b/dvc.lock
index 22e465932a78de3d6bd9adb785e97cfdc97ae467..b1e3ff4a9045f9909434ba7f5d941251150504c6 100644
--- a/dvc.lock
+++ b/dvc.lock
@@ -236,9 +236,9 @@ stages:
       size: 4617
     outs:
     - path: data/explanations/20_news/
-      md5: c8ba90f9757a4e3cc4843d3791ef2446.dir
-      size: 232912969
-      nfiles: 14041
+      md5: ad1f9f0df287078edebed1e408df2c9f.dir
+      size: 869336544
+      nfiles: 140401
   attack_basic@wiki_pl:
     cmd: PYTHONPATH=. python experiments/scripts/attack.py --dataset_name wiki_pl
       --attack_type attack_basic
diff --git a/experiments/scripts/attack.py b/experiments/scripts/attack.py
index b5e868cd82d835ad31111f891e10cfbc749d3eef..dde6b08909a48bf187ad5fec3963dcbe852ff18d 100644
--- a/experiments/scripts/attack.py
+++ b/experiments/scripts/attack.py
@@ -152,7 +152,7 @@ def load_dir_files(dir_path):
 
 def load_xai_importance(input_dir):
     global_xai_dir = os.path.join(input_dir, "global")
-    local_xai_dir = os.path.join(input_dir, "local")
+    local_xai_dir = os.path.join(input_dir, "local", "test")
     local_dirs = os.listdir(local_xai_dir)
     local_class_to_file = {dir_name: load_dir_files(os.path.join(local_xai_dir, dir_name))
                            for dir_name in local_dirs}