diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index 1111e0bbe5f626ba2cb8547e49ec88b823b0fac9..ddf10b10619387b32fafe4780e97e80fe3550ac9 100644
--- a/examples/training_navigation.py
+++ b/examples/training_navigation.py
@@ -55,9 +55,9 @@ scores = []
 dones_list = []
 action_prob = [0] * 4
 agent = Agent(state_size, action_size, "FC", 0)
-#agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint14900.pth'))
+agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint14900.pth'))
 
-demo = False
+demo = True
 
 
 def max_lt(seq, val):