diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp
new file mode 100644
index 000000000..a93506dde
Binary files /dev/null and b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp differ
diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp
index 250db9e95..c4840d9fb 100644
Binary files a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp and b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp differ
diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
index 6d3b3af8f..977668f35 100755
--- a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
+++ b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
@@ -1515,11 +1515,11 @@ def run_pea(rank, world_size, args, wb=None):
                 module.lora = module.lora.to(device)
                 module.lora = DDP(module.lora, device_ids=[rank], find_unused_parameters=False)
     
-    for i, module in enumerate(lora_modules):
-        for n, p in module.lora.named_parameters():
-            new_n = str(i) + n
-            pea_names.append(new_n)
-            pea_param.append(p)
+        for i, module in enumerate(lora_modules):
+            for n, p in module.lora.named_parameters():
+                new_n = str(i) + n
+                pea_names.append(new_n)
+                pea_param.append(p)
 
     optimizer_pea = ScaledAdam(
             pea_param,