diff --git a/amr-wind/equation_systems/tke/source_terms/KransAxell.cpp b/amr-wind/equation_systems/tke/source_terms/KransAxell.cpp
index a2c5e4e6b2..4c428df432 100644
--- a/amr-wind/equation_systems/tke/source_terms/KransAxell.cpp
+++ b/amr-wind/equation_systems/tke/source_terms/KransAxell.cpp
@@ -190,7 +190,7 @@ void KransAxell::operator()(
                     1.0 / dt * (tke_arr(i, j, k) - ref_tke);
                 src_term(i, j, k) +=
                     drag_arr(i, j, k) * terrainforcing +
-                    blank_arr(i, j, k) * dragforcing -
+                    blank_arr(i, j, k) * vf_arr(i,j,k) dragforcing -
                     static_cast<int>(has_terrain) * sponge_forcing;
                 ;
             });