diff --git a/openfl-workspace/tf_3dunet_brats/src/tf_3dunet_model.py b/openfl-workspace/tf_3dunet_brats/src/tf_3dunet_model.py
index 65d6d5f54c7..741538c7ad1 100644
--- a/openfl-workspace/tf_3dunet_brats/src/tf_3dunet_model.py
+++ b/openfl-workspace/tf_3dunet_brats/src/tf_3dunet_model.py
@@ -6,10 +6,10 @@
 import tensorflow as tf
 
 from openfl.federated import TensorFlowTaskRunner
-from .define_model import build_model
-from .define_model import dice_coef
-from .define_model import dice_loss
-from .define_model import soft_dice_coef
+from src.define_model import build_model
+from src.define_model import dice_coef
+from src.define_model import dice_loss
+from src.define_model import soft_dice_coef
 
 
 class TensorFlow3dUNet(TensorFlowTaskRunner):