diff --git a/torch_training/training_navigation.py b/torch_training/training_navigation.py
index b30278cf4865088382a352dd1ea9e242b8ef0567..b9f203e78f2a76510deb096559c56f99fe6c7556 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -210,5 +210,5 @@ for trials in range(1, n_trials + 1):
                 eps,
                 action_prob / np.sum(action_prob)))
         torch.save(agent.qnetwork_local.state_dict(),
-                   '../flatland/baselines/Nets/avoid_checkpoint' + str(trials) + '.pth')
+                   './Nets/avoid_checkpoint' + str(trials) + '.pth')
         action_prob = [1] * 4