diff --git a/tools/run_grid.py b/tools/run_grid.py
index b3ffb88..e57ae49 100644
--- a/tools/run_grid.py
+++ b/tools/run_grid.py
@@ -1,11 +1,16 @@
 import argparse
 import itertools
 import os
+import sys
 import tempfile
 
 import yaml
 import submitit
 
+PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if PROJECT_ROOT not in sys.path:
+    sys.path.insert(0, PROJECT_ROOT)
+
 from src.train_supervised import main as app_main