diff --git a/src/MNH/bl89.f90 b/src/MNH/bl89.f90
index 6aa37569a9cd0d8f7535837942d61d88dde956f1..2dc62d15f4ab73b424f2e484fff1b227eb777abd 100644
--- a/src/MNH/bl89.f90
+++ b/src/MNH/bl89.f90
@@ -319,8 +319,8 @@ ZBL89EXP = LOG(16.)/(4.*LOG(XKARMAN)+LOG(XCED)-3.*LOG(XCMFS))
 #ifdef MNH_COMPILER_NVHPC
 !$acc loop independent collapse(2)
 #endif
-do concurrent( ji = 1 : iiu, jj = 1 : iju )
-  zsqrt_tke(ji, jj) = Br_pow( ztkem(ji,jj), 0.5 )
+do concurrent( ji = 1 : iiu * iju, jk = 1 : ikt )
+  zsqrt_tke(ji, jk) = Br_pow( ztkem(ji,jk), 0.5 )
 end do
 !ZBL89EXP is defined here because (and not in ini_cturb) because XCED is defined in read_exseg (depending on BL89/RM17)
 ZBL89EXP = Br_log( 16. ) / ( 4. * Br_log( XKARMAN )+ Br_log( XCED ) - 3. * Br_log( XCMFS) )