diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
index db35c8e96..fbb4f6dd6 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp differ
diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
index 48ca16ae5..1457fcc5b 100644
Binary files a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp and b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp differ
diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/decode.py b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/decode.py
index 1390c7e44..11be96e1b 100755
--- a/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/decode.py
+++ b/egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/decode.py
@@ -825,7 +825,10 @@ def main():
     args.return_cuts = True
     tedlium = TedLiumAsrDataModule(args)
     test_cuts = tedlium.test_cuts()
-    test_dl = tedlium.train_dataloaders(train_cuts)
+    test_dl = tedlium.test_dataloaders(test_cuts)
+
+    test_sets = ['test set']
+    test_dl = [test_dl]
     
     for test_set, test_dl in zip(test_sets, test_dl):
         results_dict = decode_dataset(