diff --git a/.github/workflows/checks.yaml b/.github/workflows/checks.yaml
index f9c36cf..13a474f 100644
--- a/.github/workflows/checks.yaml
+++ b/.github/workflows/checks.yaml
@@ -41,6 +41,13 @@ jobs:
       - name: Run prediction
         run: docker run --rm -v "$(pwd)/.:/data" eyra-rank:latest /data/PreFer_fake_data.csv /data/PreFer_fake_background_data.csv --output /data/predictions.csv
 
+      - name: Check if file exists
+        run: |
+          if [ ! -f "predictions.csv" ]; then
+            echo "Predictions file not found. Please check the logs to see what went wrong."
+            exit 1
+          fi
+
       - name: Build Docker scoring image
         uses: docker/build-push-action@v4
         with:
diff --git a/python.Dockerfile b/python.Dockerfile
index 8978711..c5a3d0f 100644
--- a/python.Dockerfile
+++ b/python.Dockerfile
@@ -11,4 +11,4 @@ COPY *.py /app
 COPY *.joblib /app
 
 ENTRYPOINT ["conda", "run", "-n", "eyra-rank", "python", "/app/run.py"]
-CMD ["predict", "/data/fake_data.csv"]
\ No newline at end of file
+CMD []
\ No newline at end of file