diff --git a/Sources/restart_mod.f90 b/Sources/restart_mod.f90
index d8c54d1..068d15f 100644
--- a/Sources/restart_mod.f90
+++ b/Sources/restart_mod.f90
@@ -471,6 +471,8 @@ SUBROUTINE restart_write(restart_ext, wout_file)
       CALL cdf_define(ncid, vn_nfpin, nfp)
       CALL cdf_define(ncid, vn_wout, wout_file)
 
+      CALL cdf_define(ncid, vn_tor_modes, fourier_context%tor_modes)
+
       CALL cdf_define(ncid, vn_jbsupss, jbsupsmnsh, dimname=restart_dims)
       CALL cdf_define(ncid, vn_jbsupuc, jbsupumnch, dimname=restart_dims)
       CALL cdf_define(ncid, vn_jbsupvc, jbsupvmnch, dimname=restart_dims)
@@ -546,6 +548,8 @@ SUBROUTINE restart_write(restart_ext, wout_file)
       CALL cdf_write(ncid, vn_nfpin, nfp)
       CALL cdf_write(ncid, vn_wout, wout_file)
 
+      CALL cdf_write(ncid, vn_tor_modes, fourier_context%tor_modes)
+
       CALL cdf_write(ncid, vn_jbsupss, jbsupsmnsh)
       CALL cdf_write(ncid, vn_jbsupuc, jbsupumnch)
       CALL cdf_write(ncid, vn_jbsupvc, jbsupvmnch)