diff --git a/modelscope/models/audio/separation/m2/fsmn.py b/modelscope/models/audio/separation/m2/fsmn.py
index 97400888..07fb0f8b 100644
--- a/modelscope/models/audio/separation/m2/fsmn.py
+++ b/modelscope/models/audio/separation/m2/fsmn.py
@@ -141,4 +141,4 @@ class UniDeepFsmnDilated(nn.Module):
         out = self.conv(x_per)
         out1 = out.permute(0, 3, 2, 1)
 
-        return input + out1.squeeze()
+        return input + out1.squeeze(dim=1)