diff --git a/python/paddle/v2/fluid/tests/book_distribute/test_dist_fit_a_line.py b/python/paddle/v2/fluid/tests/book_distribute/test_dist_fit_a_line.py
index 592986b137a1023e84b4ad45d208fc8fc7670ea4..bb339c440bd0d229d2ae348cf5a7745b16d156d5 100644
--- a/python/paddle/v2/fluid/tests/book_distribute/test_dist_fit_a_line.py
+++ b/python/paddle/v2/fluid/tests/book_distribute/test_dist_fit_a_line.py
@@ -58,6 +58,5 @@ else:
                                       fetch_list=[avg_cost])
 
             if avg_loss_value[0] < 10.0:
-                exit(
-                    0)  # if avg cost less than 10.0, we think our code is good.
+                exit(0)
 exit(1)