diff --git a/neurobayes/utils/utils.py b/neurobayes/utils/utils.py
index a2dc449..57c16ed 100644
--- a/neurobayes/utils/utils.py
+++ b/neurobayes/utils/utils.py
@@ -92,8 +92,8 @@ def rmse(y_pred: jnp.ndarray, y_true: jnp.ndarray) -> jnp.ndarray:
     """
     Calculates the root mean squared error between true and predicted values.
     """
-    mse = mse(y_pred, y_true)
-    return jnp.sqrt(mse)
+    mse_ = mse(y_pred, y_true)
+    return jnp.sqrt(mse_)
 
 
 def mae(y_pred: jnp.ndarray, y_true: jnp.ndarray) -> jnp.ndarray: