diff --git a/src/physics/cam/gw_drag.F90 b/src/physics/cam/gw_drag.F90
index f04f35b8a9..6b2cdb32fd 100644
--- a/src/physics/cam/gw_drag.F90
+++ b/src/physics/cam/gw_drag.F90
@@ -1482,6 +1482,7 @@ subroutine gw_tend(state, pbuf, dt, ptend, cam_in, flx_heat)
         effgw = 0._r8
      end where
 
+     ! TODO: Once ML code is written need to modify to select which tendencies to carry forward
      if ((gw_convect_dp_ml == 'off') .or. (gw_convect_dp_ml == 'bothoff') .or. (gw_convect_dp_ml == 'bothon')) then
         ! Determine wave sources for Beres deep scheme
         call gw_beres_src(ncol, band_mid, beres_dp_desc, &
@@ -1519,6 +1520,7 @@ subroutine gw_tend(state, pbuf, dt, ptend, cam_in, flx_heat)
      taucd = calc_taucd(ncol, band_mid%ngwv, tend_level, tau, c, xv, yv, ubi)
 
      !  add the diffusion coefficients
+     ! TODO: Need to treat egwdffi if not output by ML scheme
      do k = 1, pver+1
         egwdffi_tot(:,k) = egwdffi_tot(:,k) + egwdffi(:,k)
      end do