diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index f4d8d24..6d3e1e9 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -3,6 +3,7 @@ on: push
 jobs:
   python-lint:
     runs-on: ubuntu-latest
+    container: ghcr.io/pulp-platform/pulp-nnx:main-test
     steps:
       - uses: actions/checkout@v4
       - uses: psf/black@stable