diff --git a/src/MNH/prandtl.f90 b/src/MNH/prandtl.f90
index def839a3661d1c7f15450a3059503827a1bb2458..892901e7153a936c9f598f7940f61b70339aaaad 100644
--- a/src/MNH/prandtl.f90
+++ b/src/MNH/prandtl.f90
@@ -985,7 +985,7 @@ ELSE  IF (L2D) THEN ! 3D case in a 2D model
 !$acc kernels
 !$mnh_expand_array(JI=1:JIU,JJ=1:JJU,JK=1:JKU)
 #if !defined(MNH_BITREP) && !defined(MNH_BITREP_OMP)
-      ZTMP1_DEVICE(:,:,:) = (XG / PTHVREF * PLM * PLEPS / PTKEM)**2
+      ZTMP1_DEVICE(:,:,:) = (XG / PTHVREF(:,:,:) * PLM(:,:,:) * PLEPS(:,:,:) / PTKEM(:,:,:))**2
 #else
       ZTMP1_DEVICE(:,:,:) = BR_P2(XG / PTHVREF(:,:,:) * PLM(:,:,:) * PLEPS(:,:,:) / PTKEM(:,:,:))
 #endif
@@ -1066,7 +1066,7 @@ ELSE ! 3D case in a 3D model
 !$acc kernels
 !$mnh_expand_array(JI=1:JIU,JJ=1:JJU,JK=1:JKU)
 #if !defined(MNH_BITREP) && !defined(MNH_BITREP_OMP)
-    ZTMP1_DEVICE(:,:,:) = (XG / PTHVREF * PLM * PLEPS / PTKEM)**2
+    ZTMP1_DEVICE(:,:,:) = (XG / PTHVREF(:,:,:) * PLM(:,:,:) * PLEPS(:,:,:) / PTKEM(:,:,:))**2
 #else
     ZTMP1_DEVICE(:,:,:) = BR_P2(XG / PTHVREF(:,:,:) * PLM(:,:,:) * PLEPS(:,:,:) / PTKEM(:,:,:))
 #endif