diff --git a/src/nemos/basis/_decaying_exponential.py b/src/nemos/basis/_decaying_exponential.py
index 6e867bd6..65a71a3e 100644
--- a/src/nemos/basis/_decaying_exponential.py
+++ b/src/nemos/basis/_decaying_exponential.py
@@ -154,7 +154,7 @@ def _evaluate(
 
         # process one input at the time (orthogonalization must be done one input at the time)
 
-        # flatten all inputs
+        # flatten all non-time dimensions
         shape = sample_pts.shape
         sample_pts = sample_pts.reshape(sample_pts.shape[0], -1)
         basis_list = []