diff --git a/.github/workflows/model_test_cpu.yml b/.github/workflows/model_test_cpu.yml
index 3fe43edf..ce43a9e3 100644
--- a/.github/workflows/model_test_cpu.yml
+++ b/.github/workflows/model_test_cpu.yml
@@ -34,7 +34,7 @@ jobs:
       matrix:
         include:
           - modelName: "opt-125m"
-            datasets: "piqa"
+            datasets: "lambada_openai"
             device: "cpu"
             tasks: "text-generation"
       fail-fast: true
diff --git a/.github/workflows/model_test_hpu.yml b/.github/workflows/model_test_hpu.yml
index 4914a68d..613372ef 100644
--- a/.github/workflows/model_test_hpu.yml
+++ b/.github/workflows/model_test_hpu.yml
@@ -34,7 +34,7 @@ jobs:
       matrix:
         include:
           - modelName: "opt-125m"
-            datasets: "piqa"
+            datasets: "lambada_openai"
             device: "hpu"
             tasks: "text-generation"
       fail-fast: true