diff --git a/.circleci/config.yml b/.circleci/config.yml
index 7beb7ce..8fa6777 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -12,7 +12,7 @@ jobs:
       - run:
           name: Check for Changes in Subdirectory
           command: |
-            if git diff --name-only origin/$CIRCLE_BRANCH...HEAD | grep '<< parameters.subdir >>'; then
+            if git diff --name-only HEAD~1...HEAD | grep '<< parameters.subdir >>'; then
               echo "Changes detected in << parameters.subdir >>, building Docker image."
               echo $DOCKERHUB_TOKEN | docker login --username davidwandb --password-stdin
               docker buildx create --platform linux/amd64,linux/arm64 --use
diff --git a/jobs/fashion_mnist_train/job.py b/jobs/fashion_mnist_train/job.py
index b085020..24559bd 100644
--- a/jobs/fashion_mnist_train/job.py
+++ b/jobs/fashion_mnist_train/job.py
@@ -62,6 +62,7 @@ def train(project: Optional[str], entity: Optional[str], **kwargs: Any):
         metrics=["sparse_categorical_accuracy"],
     )
     model.summary()
+
     model.fit(
         train_X,
         train_y.astype(np.float32),