diff --git a/torch_training/training_navigation.py b/torch_training/training_navigation.py
index 0c4ed8547591fe6cdd106438a4be28037baba216..1c747c27e255ea11493ad7d7cea65ed986f3bbc1 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -77,7 +77,7 @@ action_prob = [0] * action_size
 agent_obs = [None] * env.get_num_agents()
 agent_next_obs = [None] * env.get_num_agents()
 agent = Agent(state_size, action_size, "FC", 0)
-agent.qnetwork_local.load_state_dict(torch.load('./Nets/avoid_checkpoint15000.pth'))
+agent.qnetwork_local.load_state_dict(torch.load(os.path.join(__file_dirname__, 'Nets', 'avoid_checkpoint15000.pth')))
 
 demo = True