diff --git a/torchcontrib/optim/swa.py b/torchcontrib/optim/swa.py
index a980176..3ab398b 100644
--- a/torchcontrib/optim/swa.py
+++ b/torchcontrib/optim/swa.py
@@ -150,7 +150,7 @@ def _reset_lr_to_swa(self):
     def update_swa_group(self, group):
         r"""Updates the SWA running averages for the given parameter group.
 
-        Arguments:
+        Args:
             param_group (dict): Specifies for what parameter group SWA running
                 averages should be updated