diff --git a/.github/workflows/inference_diff.yml b/.github/workflows/inference_diff.yml
index 4a08519763..11a8a84733 100644
--- a/.github/workflows/inference_diff.yml
+++ b/.github/workflows/inference_diff.yml
@@ -66,7 +66,6 @@ jobs:
   post_comment:
     needs: diff_classification
     runs-on: ubuntu-latest
-    container: docker://dvcorg/cml-py3:latest
     steps:
       - uses: actions/checkout@v4
       - name: Download diff
@@ -77,6 +76,9 @@ jobs:
       - name: Prepare comment
         run: "echo \"Here's a diff of how these changes impact the classified ontology: \n\" >comment.md; cat classification-diff.md/classification-diff.md >>comment.md"
       - name: Post comment
+        uses: NejcZdovc/comment-pr@v1.1.1
         env:
-          repo_token: ${{ secrets.GITHUB_TOKEN }}
-        run: cml-send-comment comment.md
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          file: "../../comment.md"
+