diff --git a/tensorflow_transform/graph_tools.py b/tensorflow_transform/graph_tools.py
index 1221557d..178f4ab8 100644
--- a/tensorflow_transform/graph_tools.py
+++ b/tensorflow_transform/graph_tools.py
@@ -98,7 +98,7 @@ def retrieve_sources(sinks, ignore_control_dependencies=False):
   subgraph. This util is refactored from `_map_subgraph` in
   tensorflow/.../ops/op_selector.py.
 
-  Arguments:
+  Args:
     sinks:  An iterable of Operations where the subgraph terminates.
     ignore_control_dependencies: (Optional) If `True`, ignore any
       `control_inputs` for all ops while walking the graph.