diff --git a/mimir/attacks/gradnorm.py b/mimir/attacks/gradnorm.py
index e9e1c69..a1dffc5 100644
--- a/mimir/attacks/gradnorm.py
+++ b/mimir/attacks/gradnorm.py
@@ -15,7 +15,7 @@ def __init__(self, config: ExperimentConfig, model: Model):
 
     def _attack(self, document, probs, tokens=None, **kwargs):
         """
-        Gradient Norm Attack. Computes p-norm of gradients w.r.t. input tokens.
+        Gradient Norm Attack. Computes p-norm of gradients w.r.t. model parameters.
         """
         # We ignore probs here since they are computed in the general case without gradient-tracking (to save memory)