diff --git a/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py b/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
index f32e1161ad52edda50504512cfd3a482577bef03..c0f480e34dcac3351ba3008ad632a29943afdb81 100644
--- a/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
+++ b/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
@@ -22,6 +22,7 @@ os.environ[
     'RECORDIO_FILENAME'] = '/tmp/ir_memory_optimize_transformer.wmt16.recordio'
 
 from test_parallel_executor_transformer import TestTransformer
+from test_parallel_executor_transformer import transformer
 
 
 # NOTE(dzhwinter): test diferent strategy colisions.