diff --git a/src/unilab/base/torch_env.py b/src/unilab/base/torch_env.py
index a2f414f8a..cdc2b4a94 100644
--- a/src/unilab/base/torch_env.py
+++ b/src/unilab/base/torch_env.py
@@ -407,10 +407,8 @@ def _validate_state(self, state: TorchEnvState) -> None:
         self._validate_vector(steps, "info['steps']", torch.int64)
         self._validate_final_observation(state.final_observation)
         if finite_checks:
-            finite = torch.stack([torch.isfinite(value).all() for value, _ in finite_checks])
-            if not bool(finite.all()):
-                for value, label in finite_checks:
-                    self._validate_finite_float(value, label)
+            for value, label in finite_checks:
+                self._validate_finite_float(value, label)
 
     def _validate_vector(self, value: Any, label: str, dtype: torch.dtype) -> None:
         if not isinstance(value, torch.Tensor):