diff --git a/src/py/flwr/server/superlink/state/state_test.py b/src/py/flwr/server/superlink/state/state_test.py
index 3aac09d88133..4dc2203852dd 100644
--- a/src/py/flwr/server/superlink/state/state_test.py
+++ b/src/py/flwr/server/superlink/state/state_test.py
@@ -716,12 +716,14 @@ def test_store_task_res_task_ins_expired(self) -> None:
                 10,
                 time.time() - 2,
                 6,
+                True,
             ),  # TaskRes within allowed TTL
             (
                 time.time() - 5,
                 10,
                 time.time() - 2,
                 15,
+                False,
             ),  # TaskRes TTL exceeds max allowed TTL
         ]
     )
@@ -731,6 +733,7 @@ def test_store_task_res_limit_ttl(
         task_ins_ttl: float,
         task_res_created_at: float,
         task_res_ttl: float,
+        expected_store_result: bool,
     ) -> None:
         """Test the behavior of store_task_res regarding the TTL limit of TaskRes."""
         # Prepare
@@ -755,14 +758,10 @@ def test_store_task_res_limit_ttl(
         res = state.store_task_res(task_res)
 
         # Assert
-        max_allowed_ttl = (
-            task_ins.task.created_at + task_ins.task.ttl - task_res.task.created_at
-        )
-
-        if task_res.task.ttl > max_allowed_ttl:
-            assert res is None
-        else:
+        if expected_store_result:
             assert res is not None
+        else:
+            assert res is None
 
 
 def create_task_ins(