diff --git a/src/parameterizations/vertical/MOM_diabatic_driver.F90 b/src/parameterizations/vertical/MOM_diabatic_driver.F90
index bfe9c3a662..bba8f657f2 100644
--- a/src/parameterizations/vertical/MOM_diabatic_driver.F90
+++ b/src/parameterizations/vertical/MOM_diabatic_driver.F90
@@ -47,7 +47,7 @@ module MOM_diabatic_driver
 use MOM_internal_tides,      only : propagate_int_tide
 use MOM_internal_tides,      only : internal_tides_init, internal_tides_end, int_tide_CS
 use MOM_kappa_shear,         only : kappa_shear_is_used
-use MOM_KPP,                 only : KPP_CS, KPP_init, KPP_calculate, KPP_end
+use MOM_KPP,                 only : KPP_CS, KPP_init, KPP_calculate, KPP_end, KPP_get_BLD
 use MOM_KPP,                 only : KPP_NonLocalTransport_temp, KPP_NonLocalTransport_saln
 use MOM_opacity,             only : opacity_init, set_opacity, opacity_end, opacity_CS
 use MOM_regularize_layers,   only : regularize_layers, regularize_layers_init, regularize_layers_CS
@@ -637,9 +637,9 @@ subroutine diabatic(u, v, h, tv, Hml, fluxes, visc, ADp, CDp, dt, Time_end, G, G
 
     ! If visc%MLD exists, copy the KPP BLD into it
     if (associated(visc%MLD)) then
-      call pass_var(CS%KPP_CSp%OBLdepth, G%domain, halo=1)
-      visc%MLD(:,:) = CS%KPP_CSp%OBLdepth(:,:)
-      Hml(:,:) = CS%KPP_CSp%OBLdepth(:,:)
+      call KPP_get_BLD(CS%KPP_CSp, visc%MLD(:,:), G)
+      call pass_var(visc%MLD, G%domain, halo=1)
+      Hml(:,:) = visc%MLD(:,:)
     endif