diff --git a/python/paddle/v2/framework/tests/test_beam_search_op.py b/python/paddle/v2/fluid/tests/test_beam_search_op.py
similarity index 94%
rename from python/paddle/v2/framework/tests/test_beam_search_op.py
rename to python/paddle/v2/fluid/tests/test_beam_search_op.py
index a5a0cc0c96..cc7c09bb59 100644
--- a/python/paddle/v2/framework/tests/test_beam_search_op.py
+++ b/python/paddle/v2/fluid/tests/test_beam_search_op.py
@@ -1,6 +1,6 @@
 import logging
-from paddle.v2.framework.op import Operator, DynamicRecurrentOp
-import paddle.v2.framework.core as core
+from paddle.v2.fluid.op import Operator, DynamicRecurrentOp
+import paddle.v2.fluid.core as core
 import unittest
 import numpy as np