diff --git a/python/tvm/contrib/graph_runtime.py b/python/tvm/contrib/graph_runtime.py
index a960e552f68f..59db716e917c 100644
--- a/python/tvm/contrib/graph_runtime.py
+++ b/python/tvm/contrib/graph_runtime.py
@@ -141,11 +141,11 @@ class GraphModule(object):
         lib = relay.build(...)
         lib.export_library("compiled_lib.so")
         # load it back as a runtime
-        lib:tvm.runtime.Module = tvm.runtime.load_module("compiled_lib.so")
+        lib: tvm.runtime.Module = tvm.runtime.load_module("compiled_lib.so")
         # Call the library factory function for default and create
         # a new runtime.Module, wrap with graph module.
         gmod = graph_runtime.GraphModule(lib["default"](ctx))
-        # use the gmod
+        # use the graph module.
         gmod.set_input("x", data)
         gmod.run()
     """
diff --git a/python/tvm/relay/build_module.py b/python/tvm/relay/build_module.py
index 20cdc24ebc69..f05e105ed2a2 100644
--- a/python/tvm/relay/build_module.py
+++ b/python/tvm/relay/build_module.py
@@ -110,14 +110,8 @@ def build(self, mod, target=None, target_host=None, params=None):
 
         Returns
         -------
-        graph_json : str
-            The json string that can be accepted by graph runtime.
-
-        mod : tvm.Module
-            The module containing necessary libraries.
-
-        params : dict
-            The parameters of the final graph.
+        factory_module : tvm.relay.backend.graph_runtime_factory.GraphRuntimeFactoryModule
+            The runtime factory for the TVM graph runtime.
         """
         target = _update_target(target)