From 35864d4b38e5ad7efe90164845e48bf92793462c Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Tue, 13 Dec 2022 19:50:57 +0100
Subject: [PATCH] Fix: StateObservable sometimes having wrong low-bound

---
 columbus/observables.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/columbus/observables.py b/columbus/observables.py
index a903dbb..c6f34e3 100644
--- a/columbus/observables.py
+++ b/columbus/observables.py
@@ -250,7 +250,7 @@ class StateObservable(Observable):
         self.reset()
         num = len(self.entities)*2+len(self._timeoutEntities) + \
             self.speedAgent*2 + self.include_rand
-        return spaces.Box(low=0-1*self.coordsRelativeToAgent, high=1,
+        return spaces.Box(low=0-1*(self.coordsRelativeToAgent or self.speedAgent), high=1,
                           shape=(num,), dtype=np.float64)
 
     def get_observation(self):