diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 1017e237545133de527eaf369cce230c0ba1967e..8a19ecd96fbd20ea566e488d5092b065d4582721 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -9,7 +9,7 @@ test:
     - pip install -r ./deepof/requirements.txt
     - pip install -e deepof/
     - coverage run --source deepof -m pytest
-    - coverage report -m --include deepof/utils.py,deepof/preprocess.py,deepof/model_utils.py,deepof/pose_utils.py,deepof/visuals.py,deepof/models.py,deepof/hypermodels.py
+    - coverage report -m --include deepof/utils.py,deepof/preprocess.py,deepof/model_utils.py,deepof/pose_utils.py,deepof/visuals.py,deepof/models.py,deepof/hypermodels.py,deepof/train_utils.py
     - coverage xml -o deepof_cov.xml
   artifacts:
     reports:
diff --git a/deepof/models.py b/deepof/models.py
index ad7fe4f5e0256e8cdad4d9b2644dd6b199345362..f15f4a9ca9dcc526d15638175e16e6ec4095404b 100644
--- a/deepof/models.py
+++ b/deepof/models.py
@@ -315,7 +315,7 @@ class SEQ_2_SEQ_GMVAE:
                 ],
             )
 
-        else:
+        else:  # pragma: no cover
             raise NotImplementedError(
                 "Gaussian Mixtures are currently the only supported prior"
             )
diff --git a/deepof/train_utils.py b/deepof/train_utils.py
index ffa2e89aae05ddef9e9543895cd96ec5ec521346..8cd5fcb22a293677d1ac53424b49812a65efbae1 100644
--- a/deepof/train_utils.py
+++ b/deepof/train_utils.py
@@ -146,6 +146,7 @@ def tune_search(
 
     """
 
+    print(callbacks)
     tensorboard_callback, cp_callback, onecycle = callbacks
 
     if hypermodel == "S2SAE":