diff --git a/tests/test_models/phelps_deaton.py b/tests/test_models/phelps_deaton.py
index 18f1afc9..c20367ca 100644
--- a/tests/test_models/phelps_deaton.py
+++ b/tests/test_models/phelps_deaton.py
@@ -29,7 +29,9 @@ def utility_with_filter(
     consumption,
     working,
     delta,
-    lagged_retirement,  # noqa: ARG001
+    # Temporary workaround for bug described in issue #30, which requires us to pass
+    # all state variables to the utility function.
+    lagged_retirement,  # noqa: ARG001, TODO: Remove unused arguments once #30 is fixed.
 ):
     return utility(consumption=consumption, working=working, delta=delta)
 
diff --git a/tests/test_models/stochastic.py b/tests/test_models/stochastic.py
index 00eeae35..71e2f375 100644
--- a/tests/test_models/stochastic.py
+++ b/tests/test_models/stochastic.py
@@ -20,7 +20,16 @@
 # --------------------------------------------------------------------------------------
 # Utility function
 # --------------------------------------------------------------------------------------
-def utility(consumption, working, health, partner, delta, gamma):  # noqa: ARG001
+def utility(
+    consumption,
+    working,
+    health,
+    # Temporary workaround for bug described in issue #30, which requires us to pass
+    # all state variables to the utility function.
+    partner,  # noqa: ARG001, TODO: Remove unused arguments once #30 is fixed.
+    delta,
+    gamma,
+):
     return jnp.log(consumption) + (gamma * health - delta) * working