diff --git a/.github/workflows/validate.yml b/.github/workflows/validate.yml
index deec729..55f43ad 100644
--- a/.github/workflows/validate.yml
+++ b/.github/workflows/validate.yml
@@ -25,7 +25,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [blaze/macos-14, ubuntu-latest, windows-latest]
+        os: [blaze/macos-14, blaze/ubuntu-latest, windows-latest]
         rust: [stable]
         backend: ["ndarray", "wgpu", "torch"]
         include:
diff --git a/src/transformers/clip.rs b/src/transformers/clip.rs
index dcc524b..ca1ea0c 100644
--- a/src/transformers/clip.rs
+++ b/src/transformers/clip.rs
@@ -361,12 +361,6 @@ mod tests {
 
     #[test]
     fn test_init_text_embeddings() {
-        // This test is known to fail on WGPU using mesa on a Linux Github CI.
-        #[cfg(all(feature = "std", target_os = "linux", feature = "wgpu"))]
-        if std::env::var("CI").is_ok() {
-            return;
-        }
-
         let device = Default::default();
         let clip_config = ClipConfig::v1_5();
         let text_embeddings: ClipTextEmbeddings<TestBackend> =