diff --git a/bindsnet/pipeline/environment_pipeline.py b/bindsnet/pipeline/environment_pipeline.py
index 9f12a477..409da5b7 100644
--- a/bindsnet/pipeline/environment_pipeline.py
+++ b/bindsnet/pipeline/environment_pipeline.py
@@ -177,7 +177,7 @@ def env_step(self) -> Tuple[torch.Tensor, float, bool, Dict]:
                 else:
                     self.action = torch.randint(
                         low=0, high=self.env.action_space.n, size=(1,)
-                    )[0]
+                    )[0].item()
                     tqdm.write(f"too many times {self.last_action} ")
             else:
                 self.action = self.action_function(self, output=self.output)