diff --git a/funasr/models/paraformer/decoder.py b/funasr/models/paraformer/decoder.py
index f08e97b3a..b75d21dbd 100644
--- a/funasr/models/paraformer/decoder.py
+++ b/funasr/models/paraformer/decoder.py
@@ -351,9 +351,9 @@ class ParaformerSANMDecoder(BaseTransformerDecoder):
         hlens: torch.Tensor,
         ys_in_pad: torch.Tensor,
         ys_in_lens: torch.Tensor,
-        return_hidden: bool = False,
-        return_both: bool= False,
         chunk_mask: torch.Tensor = None,
+        return_hidden: bool = False,
+        return_both: bool = False,
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         """Forward decoder.
 
diff --git a/funasr/utils/misc.py b/funasr/utils/misc.py
index ef18a6146..4b58d8710 100644
--- a/funasr/utils/misc.py
+++ b/funasr/utils/misc.py
@@ -1,6 +1,7 @@
 import os
 import io
 import shutil
+import logging
 from collections import OrderedDict
 import numpy as np
 from omegaconf import DictConfig, OmegaConf