diff --git a/flow_matching/path/affine.py b/flow_matching/path/affine.py
index 81cb7ed..5b2ad5b 100644
--- a/flow_matching/path/affine.py
+++ b/flow_matching/path/affine.py
@@ -220,7 +220,7 @@ def velocity_to_epsilon(self, velocity: Tensor, x_t: Tensor, t: Tensor) -> Tenso
         return a_t * x_t + b_t * velocity
 
     def target_to_epsilon(self, x_1: Tensor, x_t: Tensor, t: Tensor) -> Tensor:
-        r"""Convert from x_1 representation to velocity.
+        r"""Convert from x_1 representation to epsilon.
 
         | given :math:`X_1`.
         | return :math:`\epsilon`.