diff --git a/vlaplex.f90 b/vlaplex.f90
index bef0e04..7df5f4e 100644
--- a/vlaplex.f90
+++ b/vlaplex.f90
@@ -228,7 +228,7 @@ program VlaPlEx
   t = 0
   call writeOutputRef()
   ! call writeOutputF(t, dt, nr, r, nv, v, f_i_old)
-  call writeOutputFCum(t, dt, r(rCum_index), nv, v, fCum_i)
+  call writeOutputFCum(t, dt, nz, r(rCum_index), nv, v, fCum_i, Z_list)
   call writeOutputPhi(t, dt, nr, r, phi, E, n_e)
   call writeOutputMom(t, dt, nz, nr, r, n_i, u_i, T_i, Z_list)
 
@@ -402,10 +402,10 @@ program VlaPlEx
     end do
     ! Write output
     if (mod(t,everyOutput) == 0 .or. t == nt) then
-      ! call writeOutputF(t, dt, nr, r, nv, v, f_i_old)
+      call writeOutputF(t, dt, nz, nr, r, nv, v, f_i_old, Z_list)
       call writeOutputPhi(t, dt, nr, r, phi, E, n_e)
       call writeOutputMom(t, dt, nz, nr, r, n_i, u_i, T_i, Z_list)
-      call writeOutputFCum(t, dt, r(rCum_index), nv, v, fCum_i(1,:))
+      call writeOutputFCum(t, dt, nz, r(rCum_index), nv, v, fCum_i, Z_list)
 
     end if