diff --git a/grid_search_configs/n_agents_grid_search/config.gin b/grid_search_configs/n_agents_grid_search/config.gin
index 430ba4509997fc05dbc42ec9e4d35171b0677937..71035d4d4059f2ee2073e8ab5a4ad72813d100d6 100644
--- a/grid_search_configs/n_agents_grid_search/config.gin
+++ b/grid_search_configs/n_agents_grid_search/config.gin
@@ -1,5 +1,6 @@
 run_grid_search.name = "n_agents_results"
 run_grid_search.num_iterations = 1002
+run_grid_search.save_every = 200
 run_grid_search.hidden_sizes = [32, 32]
 
 run_grid_search.map_width = 15
diff --git a/grid_search_train.py b/grid_search_train.py
index 808ba9f876ac713135f33f85877d919f2f948ef0..16b0a421bf119ff75bd3afa6c1a273b0e169ca24 100644
--- a/grid_search_train.py
+++ b/grid_search_train.py
@@ -119,6 +119,6 @@ def run_grid_search(name, num_iterations, n_agents, hidden_sizes, save_every,
 
 if __name__ == '__main__':
     gin.external_configurable(tune.grid_search)
-    dir = 'grid_search_configs/n_agents_grid_search'
+    dir = 'baselines/grid_search_configs/n_agents_grid_search'
     gin.parse_config_file(dir + '/config.gin')
     run_grid_search(local_dir=dir)