diff --git a/src/stokes/PressureKernels.jl b/src/stokes/PressureKernels.jl
index b42f7f901..d423c2b89 100644
--- a/src/stokes/PressureKernels.jl
+++ b/src/stokes/PressureKernels.jl
@@ -88,6 +88,25 @@ end
     return nothing
 end
 
+@parallel_indices (I...) function compute_P_kernel!(
+    P,
+    P0,
+    RP,
+    ∇V,
+    η,
+    rheology::NTuple{N,MaterialParams},
+    phase_ratio::C,
+    dt,
+    r,
+    θ_dτ,
+    ::Nothing,
+    ϕ,
+) where {N,C<:JustRelax.CellArray}
+    K = fn_ratio(get_bulk_modulus, rheology, phase_ratio[I...])
+    RP[I...], P[I...] = _compute_P!(P[I...], P0[I...], ∇V[I...], η[I...], K, dt, r, θ_dτ)
+    return nothing
+end
+
 @parallel_indices (I...) function compute_P_kernel!(
     P,
     P0,