diff --git a/maskrcnn_benchmark/modeling/rpn/rpn.py b/maskrcnn_benchmark/modeling/rpn/rpn.py
index 07997651c..c279a232f 100644
--- a/maskrcnn_benchmark/modeling/rpn/rpn.py
+++ b/maskrcnn_benchmark/modeling/rpn/rpn.py
@@ -108,8 +108,8 @@ def forward(self, x):
 
 class RPNModule(torch.nn.Module):
     """
-    Module for RPN computation. Takes feature maps from the backbone and RPN
-    proposals and losses. Works for both FPN and non-FPN.
+    Module for RPN computation. Takes feature maps from the backbone and outputs 
+    RPN proposals and losses. Works for both FPN and non-FPN.
     """
 
     def __init__(self, cfg, in_channels):