diff --git a/src/parallel_io.c b/src/parallel_io.c
index b5741fb747a0cd8a2dd9ff1645d4a22ffa780869..df8c2edcfea0f101cc051bede461c15098d95750 100644
--- a/src/parallel_io.c
+++ b/src/parallel_io.c
@@ -860,8 +860,6 @@ void prepare_file(struct engine* e, const char* baseName, long long N_total[6],
   int periodic = e->s->periodic;
   int numFiles = 1;
 
-  const struct cooling_function_data* cooling = e->cooling_func;
-
   /* First time, we need to create the XMF file */
   if (e->snapshotOutputCount == 0) xmf_create_file(baseName);
 
@@ -1084,6 +1082,7 @@ void write_output_parallel(struct engine* e, const char* baseName,
   const struct gpart* gparts = e->s->gparts;
   struct gpart* dmparts = NULL;
   const struct spart* sparts = e->s->sparts;
+  const struct cooling_function_data* cooling = e->cooling_func;
 
   /* Number of unassociated gparts */
   const size_t Ndm = Ntot > 0 ? Ntot - (Ngas + Nstars) : 0;