diff --git a/test/test_stokes_solkz.jl b/test/test_stokes_solkz.jl
index 748706c1..8d072335 100644
--- a/test/test_stokes_solkz.jl
+++ b/test/test_stokes_solkz.jl
@@ -26,8 +26,8 @@ include("../miniapps/benchmarks/stokes2D/solkz/SolKz.jl")
 function check_convergence_case1()
     nx = 64
     ny = 64
-    init_mpi = JustRelax.MPI.Initialized() ? false : true
-    _, _, iters, _ = solKz(; nx=nx, ny=ny, init_MPI=init_mpi, finalize_MPI=false)
+    init_MPI = JustRelax.MPI.Initialized() ? false : true
+    _, _, iters, _ = solKz(; nx=nx, ny=ny, init_MPI=init_MPI, finalize_MPI=false)
 
     tol = 1e-8
     passed = iters.err_evo1[end] < tol
diff --git a/test/test_stokes_solvi3D.jl b/test/test_stokes_solvi3D.jl
index 887092f1..5e24415b 100644
--- a/test/test_stokes_solvi3D.jl
+++ b/test/test_stokes_solvi3D.jl
@@ -2,7 +2,7 @@ push!(LOAD_PATH, "..")
 
 @static if ENV["JULIA_JUSTRELAX_BACKEND"] === "AMDGPU"
     using AMDGPU
-
+    AMDGPU.allowscalar(true)
 elseif ENV["JULIA_JUSTRELAX_BACKEND"] === "CUDA"
     using CUDA
     CUDA.allowscalar(true)