diff --git a/src/mflux/run.py b/src/mflux/run.py
index 4be1580..1b8e59f 100644
--- a/src/mflux/run.py
+++ b/src/mflux/run.py
@@ -3,7 +3,7 @@ import os
 import sys
 import time
 
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')))
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
 from mflux.config.model_config import ModelConfig
 from mflux.config.config import Config
diff --git a/src/mflux/save.py b/src/mflux/save.py
index 304a1e1..e34bf8f 100644
--- a/src/mflux/save.py
+++ b/src/mflux/save.py
@@ -2,7 +2,7 @@ import argparse
 import os
 import sys
 
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')))
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
 from mflux.flux.flux import Flux1
 from mflux.config.model_config import ModelConfig