diff --git a/examples/mnist-pytorch/client/train.py b/examples/mnist-pytorch/client/train.py
index 9ac9cce61..d8297de0d 100644
--- a/examples/mnist-pytorch/client/train.py
+++ b/examples/mnist-pytorch/client/train.py
@@ -5,9 +5,10 @@
 import torch
 from model import load_parameters, save_parameters
 
-from data import load_data
 from fedn.utils.helpers.helpers import save_metadata
 
+from .data import load_data
+
 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.append(os.path.abspath(dir_path))
 
diff --git a/examples/mnist-pytorch/client/validate.py b/examples/mnist-pytorch/client/validate.py
index 09328181f..e2fb097eb 100644
--- a/examples/mnist-pytorch/client/validate.py
+++ b/examples/mnist-pytorch/client/validate.py
@@ -4,9 +4,10 @@
 import torch
 from model import load_parameters
 
-from data import load_data
 from fedn.utils.helpers.helpers import save_metrics
 
+from .data import load_data
+
 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.append(os.path.abspath(dir_path))