diff --git a/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py b/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py
index 7d0c6df..de3b238 100644
--- a/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py
+++ b/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py
@@ -68,7 +68,6 @@ def forward(
         gt_affs=None,
         affs_weights=None,
         pred_affs_ac=None,
-        # affs_weights_ac=None, TODO
     ):
         self.data_dict.update(
             {