diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp
index 96703d4e8..c47e1fec9 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
index ceb91f91f..b80b2631e 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
@@ -87,6 +87,7 @@ from icefall.utils import (
     display_and_save_batch,
     setup_logger,
     str2bool,
+    save_args,
 )
 
 LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler]