diff --git a/flatland/core/env_observation_builder.py b/flatland/core/env_observation_builder.py
index 1ae2819d28170e4831a104c5149681a5e7d2fafe..6c6f84eb50d8e1c2264cf9ffba7d37ffe064b3e5 100644
--- a/flatland/core/env_observation_builder.py
+++ b/flatland/core/env_observation_builder.py
@@ -342,8 +342,8 @@ class TreeObsForRailEnv(ObservationBuilder):
 
             elif num_transitions == 0:
                 # Wrong cell type, but let's cover it and treat it as a dead-end, just in case
-                print("WRONG CELL TYPE detected in tree-search (0 transitions possible) at cell", position[0],
-                      position[1], direction)
+                #print("WRONG CELL TYPE detected in tree-search (0 transitions possible) at cell", position[0],
+                #      position[1], direction)
                 last_isTerminal = True
                 break
 
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index f1a3d20cc9873ca11b604148a8639d908ce11486..7424860b3cbc32e9c478132600d5153ed95c6365 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -1021,6 +1021,8 @@ class RailEnv(Environment):
 
                 nbits = 0
                 tmp = self.rail.get_transitions((pos[0], pos[1]))
+                print(np.sum(self.rail.get_transitions((pos[0], pos[1],direction))),self.rail.get_transitions((pos[0], pos[1],direction)),self.rail.get_transitions((pos[0], pos[1])),(pos[0], pos[1],direction))
+
                 while tmp > 0:
                     nbits += (tmp & 1)
                     tmp = tmp >> 1
diff --git a/tests/test_rendertools.py b/tests/test_rendertools.py
index 1f5c317965a3101c6232709ebb311959d5a566ed..73b42c385867ba6bc93ff794ec3ecda0bf82125d 100644
--- a/tests/test_rendertools.py
+++ b/tests/test_rendertools.py
@@ -30,8 +30,8 @@ def checkFrozenImage(sFileImage):
         if bytesFrozenImage is None:
             bytesFrozenImage = bytesImage
         else:
-            assert(bytesFrozenImage.shape == bytesImage.shape)
-            assert((np.sum(np.square(bytesFrozenImage - bytesImage)) / bytesFrozenImage.size) < 1e-3)
+            assert (bytesFrozenImage.shape == bytesImage.shape)
+            assert ((np.sum(np.square(bytesFrozenImage - bytesImage)) / bytesFrozenImage.size) < 1e-3)
 
 
 def test_render_env():