diff --git a/.github/workflows/build-ci-docker-images.yml b/.github/workflows/build-ci-docker-images.yml
index 9fba7a07086ea0..a0d7c67a6cf6b6 100644
--- a/.github/workflows/build-ci-docker-images.yml
+++ b/.github/workflows/build-ci-docker-images.yml
@@ -26,7 +26,7 @@ jobs:
 
     strategy:
       matrix:
-        file: ["quality", "consistency", "custom-tokenizers", "torch-light", "tf-light", "exotic-models", "torch-tf-light", "torch-flax-light", "flax-light"]
+        file: ["quality", "consistency", "custom-tokenizers", "torch-light", "tf-light", "exotic-models", "torch-tf-light", "torch-jax-light", "jax-light"]
     continue-on-error: true 
 
     steps:
diff --git a/docker/flax-light.dockerfile b/docker/jax-light.dockerfile
similarity index 100%
rename from docker/flax-light.dockerfile
rename to docker/jax-light.dockerfile
diff --git a/docker/torch-flax-light.dockerfile b/docker/torch-jax-light.dockerfile
similarity index 100%
rename from docker/torch-flax-light.dockerfile
rename to docker/torch-jax-light.dockerfile