diff --git a/.github/workflows/publish_images.yml b/.github/workflows/publish_images.yml
index 63717276..3abaac41 100644
--- a/.github/workflows/publish_images.yml
+++ b/.github/workflows/publish_images.yml
@@ -88,7 +88,7 @@ jobs:
         with:
           context: .
           push: true
-          tags: ${{ env.image_name }}:latest
+          tags: ${{ env.REGISTRY }}/${{ env.image_name }}:latest
           build-args: |
             python_version=${{ matrix.python_version }}
             target=${{ matrix.target }}
diff --git a/Dockerfile b/Dockerfile
index 0db19548..ab4379bd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -39,7 +39,6 @@ RUN  if [ "$target" = "hannah" ] || [ "$target" = "mlonmcu" ]; then\
         apt-get update -y && apt-get -y install git mesa-utils  libblas-dev liblapack-dev libsndfile1-dev libsox-dev cmake ninja-build curl build-essential;  \
       fi
 
-RUN echo "Target: $target"
 
 # Install poetry using recommended method
 RUN  pip install poetry
diff --git a/hannah/train.py b/hannah/train.py
index 0d89ad32..3297a443 100644
--- a/hannah/train.py
+++ b/hannah/train.py
@@ -120,9 +120,6 @@ def train(
         logger.append(CSVLogger(".", version="logs", name=""))
         logger.append(JSONLogger(".", version="logs", name=""))
 
-        # if DVCLIVE_AVAILABLE:
-        #    logger.append(DVCLogger())
-
         callbacks = []
 
         callbacks.extend(list(common_callbacks(config)))