diff --git a/src/mednet/scripts/utils.py b/src/mednet/scripts/utils.py
index b1adfb932c9865b83f6383248b09dc4ca2ee2e57..4c894bd708c0db4a4ce464dcd243b90c0a774b3f 100644
--- a/src/mednet/scripts/utils.py
+++ b/src/mednet/scripts/utils.py
@@ -12,8 +12,7 @@ import shutil
 import lightning.pytorch
 import lightning.pytorch.callbacks
 import torch.nn
-
-from ..engine.device import SupportedPytorchDevice
+from medbase.engine.device import SupportedPytorchDevice
 
 logger = logging.getLogger(__name__)
 
@@ -59,7 +58,11 @@ def device_properties(
         Static properties of the current machine.
     """
 
-    from ..utils.resources import cpu_constants, cuda_constants, mps_constants
+    from medbase.utils.resources import (
+        cpu_constants,
+        cuda_constants,
+        mps_constants,
+    )
 
     retval: dict[str, int | float | str] = {}
     retval.update(cpu_constants())