diff --git a/demos/streaming_tts_server/test_client.sh b/demos/streaming_tts_server/test_client.sh
index 333ae00d..86982095 100644
--- a/demos/streaming_tts_server/test_client.sh
+++ b/demos/streaming_tts_server/test_client.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # http client test
-paddlespeech_client tts --server_ip 127.0.0.1 --port 8092 --protocol http --input "您好，欢迎使用百度飞桨语音合成服务。" --output output.wav
+paddlespeech_client tts_online --server_ip 127.0.0.1 --port 8092 --protocol http --input "您好，欢迎使用百度飞桨语音合成服务。" --output output.wav
 
 # websocket client test
-#paddlespeech_client tts --server_ip 127.0.0.1 --port 8092 --protocol websocket --input "您好，欢迎使用百度飞桨语音合成服务。" --output output.wav
\ No newline at end of file
+#paddlespeech_client tts_online --server_ip 127.0.0.1 --port 8092 --protocol websocket --input "您好，欢迎使用百度飞桨语音合成服务。" --output output.wav
diff --git a/paddlespeech/t2s/modules/predictor/length_regulator.py b/paddlespeech/t2s/modules/predictor/length_regulator.py
index b64aa44a..11f28408 100644
--- a/paddlespeech/t2s/modules/predictor/length_regulator.py
+++ b/paddlespeech/t2s/modules/predictor/length_regulator.py
@@ -49,7 +49,9 @@ class LengthRegulator(nn.Layer):
         encodings: (B, T, C)
         durations: (B, T)
         """
-        batch_size, t_enc = durations.shape
+        #batch_size, t_enc = durations.shape
+        batch_size = paddle.shape(durations)[0]
+        t_enc = paddle.shape(durations)[1]
         durations = durations.numpy()
         slens = np.sum(durations, -1)
         t_dec = np.max(slens)