diff --git a/.github/workflows/ci-gpu-rdna.yaml b/.github/workflows/ci-gpu-rdna.yaml
index c1abc7def..5660e10de 100644
--- a/.github/workflows/ci-gpu-rdna.yaml
+++ b/.github/workflows/ci-gpu-rdna.yaml
@@ -81,7 +81,7 @@ jobs:
         run: |
           # Install User libraries
           sudo apt-get update
-          sudo apt install -y libnuma1 numactl gfortran build-essential binutils dwarfdump
+          sudo apt install -y dwarfdump
 
           # Install torch+rocm6.4
           python -m pip install --upgrade pip
diff --git a/.github/workflows/ci-gpu.yaml b/.github/workflows/ci-gpu.yaml
index 232870423..43cfbc141 100644
--- a/.github/workflows/ci-gpu.yaml
+++ b/.github/workflows/ci-gpu.yaml
@@ -59,8 +59,8 @@ jobs:
       - name: Setup env
         if: steps.cache-llvm-mlir.outputs.cache-hit != 'true'
         run: |
-          sudo apt update
-          sudo apt install -y ninja-build cmake clang lld dwarfdump
+          sudo apt-get update
+          sudo apt-get install -y ninja-build cmake clang lld dwarfdump
           pip install -r water/requirements-dev.txt
 
       - name: Checkout LLVM
@@ -276,8 +276,8 @@ jobs:
 
       - name: Setup env
         run: |
-          sudo apt update
-          sudo apt install -y ninja-build cmake clang lld
+          sudo apt-get update
+          sudo apt-get install -y ninja-build cmake clang lld
           pip install -r water/requirements-dev.txt
 
       - name: Setup Cache Vars
diff --git a/tests/kernel/wave_gemm_test.py b/tests/kernel/wave_gemm_test.py
index cdd10bf45..95b33543f 100644
--- a/tests/kernel/wave_gemm_test.py
+++ b/tests/kernel/wave_gemm_test.py
@@ -111,7 +111,7 @@ def get_test_shapes(test_name: str) -> list[tuple[int]]:
     ],
 )
 def testGemmBench(tmp_path, mfma_variant: MMAType, threads_per_wave: int):
-    shape = (64, 64, 64)
+    shape = (128, 128, 128)
     perf_filename_tk = tmp_path / "wave_gemm_bench.txt"
     perf_filename_iree = tmp_path / "iree_gemm_bench.txt"
     enable_scheduling = SchedulingType.NONE