diff --git a/text.py b/text.py
index 1ad66a07ead3b3588c94382d2c8ec9069527c190..2702981d6e274ab20e250b7499b65632ffd7a3ba 100644
--- a/text.py
+++ b/text.py
@@ -13,7 +13,7 @@ from paddle.fluid.data_feeder import convert_dtype
 
 from paddle.fluid import layers
 from paddle.fluid.dygraph import Layer
-from paddle.layers import BeamSearchDecoder
+from paddle.fluid.layers import BeamSearchDecoder
 
 __all__ = [
     'RNNCell', 'BasicLSTMCell', 'BasicGRUCell', 'RNN', 'DynamicDecode',