diff --git a/crates/burn-core/src/nn/loss/huber.rs b/crates/burn-core/src/nn/loss/huber.rs
index 8b227b0a4..55d57d65c 100644
--- a/crates/burn-core/src/nn/loss/huber.rs
+++ b/crates/burn-core/src/nn/loss/huber.rs
@@ -37,7 +37,7 @@ impl HuberLossConfig {
 /// The loss for each element of the residuals `r = targets - predictions` is given by
 ///
 /// ```text
-/// L(r) = 0.5 * x^2                  if |r| <= d
+/// L(r) = 0.5 * r^2                  if |r| <= d
 /// L(r) = 0.5 * d^2 + d * (|r| - d)  if |r| >  d
 /// ```
 ///