diff --git a/src/MNH/prandtl.f90 b/src/MNH/prandtl.f90
index 9c1fe36274d6f7d0d55ca9150aa7c061bf511dc4..def839a3661d1c7f15450a3059503827a1bb2458 100644
--- a/src/MNH/prandtl.f90
+++ b/src/MNH/prandtl.f90
@@ -759,8 +759,8 @@ ELSE                                 ! 3D case in a 3D model
                                 + BR_P2( XCTV*PBLL_O_E(:,:,:)*PETHETA(:,:,:) ) &
                                 * ZTMP2_DEVICE(:,:,:)
 #endif
-    PRED2TH3(:,:,IKB)=PRED2TH3(:,:,IKB+KKL)
 !$mnh_end_expand_array()
+    PRED2TH3(:,:,IKB)=PRED2TH3(:,:,IKB+KKL)
 !$acc end kernels
 #endif
 !