diff --git a/ml-agents/mlagents/trainers/torch/utils.py b/ml-agents/mlagents/trainers/torch/utils.py
index 600c7eb8d9..c459c9ef79 100644
--- a/ml-agents/mlagents/trainers/torch/utils.py
+++ b/ml-agents/mlagents/trainers/torch/utils.py
@@ -162,5 +162,5 @@ def get_probs_and_entropy(
             entropies = entropies.squeeze(-1)
             all_probs = None
         else:
-            all_probs = torch.cat(all_probs, dim=-1)
+            all_probs = torch.cat(all_probs_list, dim=-1)
         return log_probs, entropies, all_probs