diff --git a/python/paddle/fluid/contrib/mixed_precision/bf16/amp_utils.py b/python/paddle/fluid/contrib/mixed_precision/bf16/amp_utils.py
index 8ee5cb5629ce3..0fb86593b2d62 100644
--- a/python/paddle/fluid/contrib/mixed_precision/bf16/amp_utils.py
+++ b/python/paddle/fluid/contrib/mixed_precision/bf16/amp_utils.py
@@ -128,8 +128,8 @@ def _insert_cast_op(block, op, idx, src_dtype, dest_dtype):
                     continue
                 if out_var.dtype == core.VarDesc.VarType.FP32:
                     out_var.desc.set_dtype(core.VarDesc.VarType.BF16)
-                if op.has_attr('out_dtype'):
-                    op._set_attr('out_dtype', core.VarDesc.VarType.BF16)
+                    if op.has_attr('out_dtype'):
+                        op._set_attr('out_dtype', core.VarDesc.VarType.BF16)
     return num_cast_ops
 
 
diff --git a/python/paddle/fluid/tests/book/CMakeLists.txt b/python/paddle/fluid/tests/book/CMakeLists.txt
index d8cadae068a38..09c650f16e2fb 100644
--- a/python/paddle/fluid/tests/book/CMakeLists.txt
+++ b/python/paddle/fluid/tests/book/CMakeLists.txt
@@ -12,4 +12,4 @@ set_tests_properties(test_image_classification PROPERTIES TIMEOUT 200)
 set_tests_properties(test_label_semantic_roles PROPERTIES TIMEOUT 240)
 set_tests_properties(test_machine_translation PROPERTIES TIMEOUT 120)
 set_tests_properties(test_rnn_encoder_decoder PROPERTIES TIMEOUT 120)
-set_tests_properties(test_fit_a_line PROPERTIES TIMEOUT 140)
+set_tests_properties(test_fit_a_line PROPERTIES TIMEOUT 120)