diff --git a/src/simulation/m_rhs.fpp b/src/simulation/m_rhs.fpp
index 4ee5cb402f..ce297b823d 100644
--- a/src/simulation/m_rhs.fpp
+++ b/src/simulation/m_rhs.fpp
@@ -1011,8 +1011,8 @@ contains
         if (run_time_info .or. probe_wrt .or. ib) then
 
             ix%beg = -buff_size; iy%beg = 0; iz%beg = 0
-            if (n > 0) iy%beg = -buff_size;
-            if (p > 0) iz%beg = -buff_size;
+            if (n > 0) iy%beg = -buff_size; 
+            if (p > 0) iz%beg = -buff_size; 
             ix%end = m - ix%beg; iy%end = n - iy%beg; iz%end = p - iz%beg
             !$acc update device(ix, iy, iz)