diff --git a/src/parallel_io.c b/src/parallel_io.c
index 073a5c26325dee7d0a5cb7bf9b4367476dc53ae9..2aa7a9c49642df47cc56149a745fe29f36659426 100644
--- a/src/parallel_io.c
+++ b/src/parallel_io.c
@@ -955,6 +955,8 @@ void prepare_file(struct engine* e, const char* baseName, long long N_total[6],
   const struct spart* sparts = e->s->sparts;
   struct swift_params* params = e->parameter_file;
   const int with_cosmology = e->policy & engine_policy_cosmology;
+  const int with_cooling = e->policy & engine_policy_cooling;
+  const int with_temperature = e->policy & engine_policy_temperature;
 
   FILE* xmfFile = 0;
   int numFiles = 1;
@@ -1219,6 +1221,8 @@ void write_output_parallel(struct engine* e, const char* baseName,
   const struct spart* sparts = e->s->sparts;
   struct swift_params* params = e->parameter_file;
   const int with_cosmology = e->policy & engine_policy_cosmology;
+  const int with_cooling = e->policy & engine_policy_cooling;
+  const int with_temperature = e->policy & engine_policy_temperature;
 
   /* Number of particles currently in the arrays */
   const size_t Ntot = e->s->nr_gparts;