diff --git a/datamint/lightning/trainers/specialized/nnunet/trainer.py b/datamint/lightning/trainers/specialized/nnunet/trainer.py
index 65fc90b7..c2b62ccf 100644
--- a/datamint/lightning/trainers/specialized/nnunet/trainer.py
+++ b/datamint/lightning/trainers/specialized/nnunet/trainer.py
@@ -549,4 +549,5 @@ def _build_deploy_adapter(self, dataset_id: int, bridge) -> None:
                 name='nnunet_model',
                 artifacts={'nnunet_bundle': str(bundle)},
                 extra_pip_requirements=[f'nnunetv2=={_nnunet_ver}'],
+                annotation_specs=self._build_annotation_specs(),
             )