diff --git a/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py b/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py
index 2162989cfbf9d54bd862f2427bb4ccb00789ea4a..93a7215410b9fee00c51367f3036f599ecf8eb96 100644
--- a/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py
+++ b/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py
@@ -109,6 +109,7 @@ def train(use_cuda, train_program, parallel, params_dirname):
             feed_order=['pixel', 'label'])
     else:
         import paddle.fluid.core as core
+        import paddle.compat as cpt
         try:
             trainer.train(
                 reader=train_reader,
diff --git a/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py b/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py
index f318964346532b9947ed1c143fa7a69b167dc3f8..d3e42446214ad2c86d0105747280fb017774609b 100644
--- a/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py
+++ b/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py
@@ -92,6 +92,7 @@ def train(use_cuda, train_program, params_dirname, parallel):
             feed_order=['img', 'label'])
     else:
         import paddle.fluid.core as core
+        import paddle.compat as cpt
         try:
             trainer.train(
                 num_epochs=1,