diff --git a/src/main.f90 b/src/main.f90
index db21e2fa..a7d3d11d 100644
--- a/src/main.f90
+++ b/src/main.f90
@@ -250,7 +250,7 @@ program cans
   grid_vol_ratio_c(:) = dl(1)*dl(2)*dzc(:)/(l(1)*l(2)*l(3))
   grid_vol_ratio_f(:) = dl(1)*dl(2)*dzf(:)/(l(1)*l(2)*l(3))
   !$acc end kernels
-  !$acc update self(dzci,dzfi) async
+  !$acc update self(zc,zf,dzc,dzf,dzci,dzfi) async
   !$acc exit data copyout(zc_g,zf_g,dzc_g,dzf_g,dzci_g,dzfi_g) async ! not needed on the device
   !$acc wait
   !
@@ -309,7 +309,6 @@ program cans
   if(.not.restart) then
     istep = 0
     time = 0.
-    !$acc update self(zc,dzc,dzf)
     call initflow(inivel,bcvel,ng,lo,l,dl,zc,zf,dzc,dzf,visc, &
                   is_forced,velf,bforce,is_wallturb,u,v,w,p)
     if(myid == 0) print*, '*** Initial condition succesfully set ***'