From 0076e38be0f20d47f2be4bb0ae2ae57565be9fbd Mon Sep 17 00:00:00 2001
From: Zhaoyi Shen <11598433+szy21@users.noreply.github.com>
Date: Fri, 29 Mar 2024 13:25:59 -0700
Subject: [PATCH] fix

---
 src/prognostic_equations/implicit/implicit_solver.jl | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/prognostic_equations/implicit/implicit_solver.jl b/src/prognostic_equations/implicit/implicit_solver.jl
index c3f98b91b99..1e65bcc3574 100644
--- a/src/prognostic_equations/implicit/implicit_solver.jl
+++ b/src/prognostic_equations/implicit/implicit_solver.jl
@@ -408,12 +408,16 @@ NVTX.@annotate function Wfact!(A, Y, p, dtγ, t)
             p.atmos.turbconv_model isa AbstractEDMF ?
             (; p.precomputed.ᶜtke⁰, p.precomputed.ᶜmixing_length) : (;)
         )...,
+        (
+            use_derivative(A.diffusion_flag) &&
+            p.atmos.turbconv_model isa PrognosticEDMFX ?
+            (; p.precomputed.ᶜρa⁰) : (;)
+        )...,
         (
             use_derivative(A.sgs_advection_flag) &&
             p.atmos.turbconv_model isa PrognosticEDMFX ?
             (;
                 p.core.ᶜgradᵥ_ᶠΦ,
-                p.precomputed.ᶜρa⁰,
                 p.precomputed.ᶜρʲs,
                 p.precomputed.ᶠu³ʲs,
                 p.precomputed.ᶜtsʲs,