diff --git a/doc/fluid/api/gen_doc.py b/doc/fluid/api/gen_doc.py
index 65e92c908b471de7adc9e7281780beea0f792cde..507a615ec3980d93692308246ea7ff07dbbba7e5 100644
--- a/doc/fluid/api/gen_doc.py
+++ b/doc/fluid/api/gen_doc.py
@@ -18,6 +18,7 @@ import sys
 import types
 import os
 import contextlib
+import paddle.fleet as fleet
 import paddle.fluid as fluid
 import paddle.tensor as tensor
 import paddle.nn as nn