1 files changed, 24 insertions, 0 deletions
diff --git a/schedule.ccl b/schedule.ccl
index 4799da3..8579a96 100755
--- a/schedule.ccl
+++ b/schedule.ccl
@@ -14,6 +14,14 @@ if (timelevels == 1)
   {
     STORAGE: Bvec[1]
   }
+  if (!CCTK_EQUALS(initial_temperature, "none"))
+  {
+    STORAGE: temperature[1]
+  }
+  if (!CCTK_EQUALS(initial_entropy, "none"))
+  {
+    STORAGE: entropy[1]
+  }
 }
 else if (timelevels == 2)
 {
@@ -29,6 +37,14 @@ else if (timelevels == 2)
   {
     STORAGE: Bvec[2]
   }
+  if (!CCTK_EQUALS(initial_temperature, "none"))
+  {
+    STORAGE: temperature[2]
+  }
+  if (!CCTK_EQUALS(initial_entropy, "none"))
+  {
+    STORAGE: entropy[2]
+  }
 }
 else if (timelevels == 3)
 {
@@ -44,6 +60,14 @@ else if (timelevels == 3)
   {
     STORAGE: Bvec[3]
   }
+  if (!CCTK_EQUALS(initial_temperature, "none"))
+  {
+    STORAGE: temperature[3]
+  }
+  if (!CCTK_EQUALS(initial_entropy, "none"))
+  {
+    STORAGE: entropy[3]
+  }
 }
 
 if (hydro_excision)