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 2767e8b5d948da4aee1010b99f1a58435f81fae2..d4e6742b7826e9edf92dc841ee30ae04e9309e47 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
@@ -107,12 +107,6 @@ def train(use_cuda, train_program, parallel, params_dirname):
         event_handler=event_handler,
         feed_order=['pixel', 'label'])
 
-    def _del_trainer(trainer):
-        del trainer
-
-    if six.PY3:
-        _del_trainer(trainer)
-
 
 def infer(use_cuda, inference_program, parallel, params_dirname=None):
     place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
@@ -158,4 +152,5 @@ if __name__ == '__main__':
         for parallel in (False, True):
             if use_cuda and not core.is_compiled_with_cuda():
                 continue
-            main(use_cuda=use_cuda, parallel=parallel)
+            if six.PY2:
+                main(use_cuda=use_cuda, parallel=parallel)
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 b7846574664f9190ce5c1fd52771f8b572425aab..811f17b5ab36dadb84cdf2ad43f026667266d9f0 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
@@ -90,12 +90,6 @@ def train(use_cuda, train_program, params_dirname, parallel):
         reader=train_reader,
         feed_order=['img', 'label'])
 
-    def _del_trainer(trainer):
-        del trainer
-
-    if six.PY3:
-        _del_trainer(trainer)
-
 
 def infer(use_cuda, inference_program, parallel, params_dirname=None):
     place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
@@ -143,4 +137,5 @@ if __name__ == '__main__':
         for parallel in (False, True):
             if use_cuda and not core.is_compiled_with_cuda():
                 continue
-            main(use_cuda=use_cuda, parallel=parallel)
+            if six.PY2:
+                main(use_cuda=use_cuda, parallel=parallel)