diff --git a/src/anemoi/inference/checkpoint/__init__.py b/src/anemoi/inference/checkpoint/__init__.py
index cc709a4..9159289 100644
--- a/src/anemoi/inference/checkpoint/__init__.py
+++ b/src/anemoi/inference/checkpoint/__init__.py
@@ -17,9 +17,9 @@
 from anemoi.utils.provenance import gather_provenance_info
 from packaging.version import Version
 
-from .metadata import Metadata
-from .package_exemptions import EXEMPT_NAMESPACES
-from .package_exemptions import EXEMPT_PACKAGES
+from anemoi.inference.checkpoint.metadata import Metadata
+from anemoi.inference.checkpoint.package_exemptions import EXEMPT_NAMESPACES
+from anemoi.inference.checkpoint.package_exemptions import EXEMPT_PACKAGES
 
 LOG = logging.getLogger(__name__)