diff --git a/src/diagnose.F90 b/src/diagnose.F90
index c8e2df7ee4bb43cff2bcb4fed2982aba61f48927..f3f286152acc7647ad98f57e7fac8cc7381dbbfb 100644
--- a/src/diagnose.F90
+++ b/src/diagnose.F90
@@ -273,7 +273,7 @@ SUBROUTINE diagnose(kstep)
      !   Close all diagnostic files
      CALL closef(fidres)
      CALL closef(fidrst)
-     
+
   END IF
 
 END SUBROUTINE diagnose
@@ -442,10 +442,10 @@ SUBROUTINE checkpoint_save(cp_step)
   WRITE(dset_name, "(A, '/', i6.6)") "/Basic/moments_e", cp_step
   IF (num_procs .EQ. 1) THEN
     CALL putarr(fidrst, dset_name, moments_e(ips_e:ipe_e,ijs_e:ije_e,&
-                                                      ikrs:ikre,ikzs:ikze,1), ionode=0)
+                                                      ikrs:ikre,ikzs:ikze,updatetlevel), ionode=0)
   ELSE
     CALL putarr(fidrst, dset_name, moments_e(ips_e:ipe_e,ijs_e:ije_e,&
-                                                      ikrs:ikre,ikzs:ikze,1), pardim=3)
+                                                      ikrs:ikre,ikzs:ikze,updatetlevel), pardim=3)
   ENDIF
 
   CALL attach(fidrst, dset_name, 'cstep', cstep)
@@ -458,10 +458,10 @@ SUBROUTINE checkpoint_save(cp_step)
   WRITE(dset_name, "(A, '/', i6.6)") "/Basic/moments_i", cp_step
   IF (num_procs .EQ. 1) THEN
     CALL putarr(fidrst, dset_name, moments_i(ips_i:ipe_i,ijs_i:ije_i,&
-                                                      ikrs:ikre,ikzs:ikze,1), ionode=0)
+                                                      ikrs:ikre,ikzs:ikze,updatetlevel), ionode=0)
   ELSE
     CALL putarr(fidrst, dset_name, moments_i(ips_i:ipe_i,ijs_i:ije_i,&
-                                                      ikrs:ikre,ikzs:ikze,1), pardim=3)
+                                                      ikrs:ikre,ikzs:ikze,updatetlevel), pardim=3)
   ENDIF
 
   CALL attach(fidrst, dset_name, 'cstep', cstep)
diff --git a/src/inital.F90 b/src/inital.F90
index 916e6c878f67c593eb1aed39bd5159ca8863b36d..5ad14fcd8f61b041d1da4fed2b3dc72fe3429963 100644
--- a/src/inital.F90
+++ b/src/inital.F90
@@ -127,7 +127,7 @@ SUBROUTINE load_cp
   WRITE(*,'(3x,a)') "Resume from previous run"
 
   CALL openf(rstfile, fidrst,mpicomm=MPI_COMM_WORLD)
-  
+
     n_ = 0
     WRITE(dset_name, "(A, '/', i6.6)") "/Basic/moments_e", n_
     DO WHILE (isdataset(fidrst, dset_name))
@@ -139,8 +139,8 @@ SUBROUTINE load_cp
     WRITE(dset_name, "(A, '/', i6.6)") "/Basic/moments_i", n_
 
     ! Read state of system from restart file
-    CALL getarr(fidrst, dset_name, moments_i(ips_i:ipe_i,ijs_i:ije_i,ikrs:ikre,ikzs:ikze,1),pardim=3)
-    CALL getarr(fidrst, dset_name, moments_e(ips_e:ipe_e,ijs_e:ije_e,ikrs:ikre,ikzs:ikze,1),pardim=3)
+    CALL getarr(fidrst, dset_name, moments_i(ips_i:ipe_i,ijs_i:ije_i,ikrs:ikre,ikzs:ikze,updatetlevel),pardim=3)
+    CALL getarr(fidrst, dset_name, moments_e(ips_e:ipe_e,ijs_e:ije_e,ikrs:ikre,ikzs:ikze,updatetlevel),pardim=3)
 
     ! Read time dependent attributes
     CALL getatt(fidrst, dset_name, 'cstep', cstep)