diff --git a/examples/simple_example_1.py b/examples/simple_example_1.py
index 93d5268840c4980aea40ae3a18b35f2bed31260e..7956c34fd4a5b94859a4b64441450afe2114133c 100644
--- a/examples/simple_example_1.py
+++ b/examples/simple_example_1.py
@@ -1,5 +1,4 @@
 from flatland.envs.generators import rail_from_manual_specifications_generator
-from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
 
@@ -13,12 +12,11 @@ specs = [[(0, 0), (0, 0), (0, 0), (0, 0), (0, 0), (0, 0)],
 env = RailEnv(width=6,
               height=4,
               rail_generator=rail_from_manual_specifications_generator(specs),
-              number_of_agents=1,
-              obs_builder_object=TreeObsForRailEnv(max_depth=2))
+              number_of_agents=1)
 
 env.reset()
 
 env_renderer = RenderTool(env)
-env_renderer.render_env(show=True)
+env_renderer.render_env(show=True, show_predictions=False, show_observations=False)
 
 input("Press Enter to continue...")
diff --git a/examples/simple_example_2.py b/examples/simple_example_2.py
index f8612a392d0c6d57f6077dc988379e927af1e1b1..994c7deda1569b77d4adac8a17fa9ebe14b27ef6 100644
--- a/examples/simple_example_2.py
+++ b/examples/simple_example_2.py
@@ -3,7 +3,6 @@ import random
 import numpy as np
 
 from flatland.envs.generators import random_rail_generator
-from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
 
@@ -27,8 +26,7 @@ transition_probability = [1.0,  # empty cell - Case 0
 env = RailEnv(width=10,
               height=10,
               rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
-              number_of_agents=3,
-              obs_builder_object=TreeObsForRailEnv(max_depth=2))
+              number_of_agents=3)
 
 env.reset()