diff --git a/src/simulation/m_data_output.fpp b/src/simulation/m_data_output.fpp
index 255e3a4dd3..a05b62d5a5 100644
--- a/src/simulation/m_data_output.fpp
+++ b/src/simulation/m_data_output.fpp
@@ -1062,6 +1062,8 @@ contains
 
         integer, intent(in) :: time_step
 
+        $:GPU_UPDATE(host='[patch_ib(1:num_ibs)]')
+
         if (parallel_io) then
             call s_write_parallel_ib_state(time_step)
         else
diff --git a/src/simulation/m_ibm.fpp b/src/simulation/m_ibm.fpp
index 18eebf998e..811db7e54f 100644
--- a/src/simulation/m_ibm.fpp
+++ b/src/simulation/m_ibm.fpp
@@ -935,10 +935,10 @@ contains
             end do
         end if
 
-        $:GPU_PARALLEL_LOOP(private='[ib_idx, ib_idx_temp, encoded_ib_idx, fluid_idx, radial_vector, local_force_contribution, &
-                            & cell_volume, local_torque_contribution, dynamic_viscosity, viscous_stress_div, &
-                            & viscous_stress_div_1, viscous_stress_div_2, dx, dy, dz]', copy='[forces, torques]', &
-                            & copyin='[dynamic_viscosities]', collapse=3)
+        $:GPU_PARALLEL_LOOP(private='[i, j, k, l, ib_idx, ib_idx_temp, encoded_ib_idx, fluid_idx, radial_vector, &
+                            & local_force_contribution, cell_volume, local_torque_contribution, dynamic_viscosity, &
+                            & viscous_stress_div, viscous_stress_div_1, viscous_stress_div_2, dx, dy, dz]', copy='[forces, &
+                            & torques]', copyin='[dynamic_viscosities]', collapse=3)
         do i = 0, m
             do j = 0, n
                 do k = 0, p