diff --git a/src/cooling/Compton/cooling.h b/src/cooling/Compton/cooling.h
index 632e6e07b669dc8d847a35ab171b991e20c735ae..f440cd03455c07d2eeb64c37189aed36efe78e09 100644
--- a/src/cooling/Compton/cooling.h
+++ b/src/cooling/Compton/cooling.h
@@ -39,10 +39,9 @@
  * @brief Common operations performed on the cooling function at a
  * given time-step or redshift.
  *
- * @param phys_const The physical constants in internal units.
- * @param us The internal system of units.
  * @param cosmo The current cosmological model.
  * @param cooling The #cooling_function_data used in the run.
+ * @param restart_flag Are we calling this directly after a restart?
  */
 INLINE static void cooling_update(const struct cosmology* cosmo,
                                   struct cooling_function_data* cooling,
diff --git a/src/cooling/EAGLE/cooling_tables.c b/src/cooling/EAGLE/cooling_tables.c
index 1da1f1c0140e724133d2a80d47474a6cc93c0e55..4f3aed05746ee0f7f6559936ad151d8227bbb90c 100644
--- a/src/cooling/EAGLE/cooling_tables.c
+++ b/src/cooling/EAGLE/cooling_tables.c
@@ -18,7 +18,7 @@
  ******************************************************************************/
 
 /**
- * @file src/cooling/EAGLE/eagle_cool_tables.c
+ * @file src/cooling/EAGLE/cooling_tables.c
  * @brief Functions to read EAGLE tables
  */
 
diff --git a/src/cooling/const_du/cooling.h b/src/cooling/const_du/cooling.h
index eb9b098105b2f58baa3eadd2d14994b451da538d..dac92f09837cbb40cc49f1e8dc5d4c627ce7023a 100644
--- a/src/cooling/const_du/cooling.h
+++ b/src/cooling/const_du/cooling.h
@@ -49,10 +49,9 @@
  * @brief Common operations performed on the cooling function at a
  * given time-step or redshift.
  *
- * @param phys_const The physical constants in internal units.
- * @param us The internal system of units.
  * @param cosmo The current cosmological model.
  * @param cooling The #cooling_function_data used in the run.
+ * @param restart_flag Are we calling this directly after a restart?
  */
 INLINE static void cooling_update(const struct cosmology* cosmo,
                                   struct cooling_function_data* cooling,
diff --git a/src/cooling/const_lambda/cooling.h b/src/cooling/const_lambda/cooling.h
index eef62330ae9b52314ad0135b5390f9f31eb239e5..3c336060bdadae9b0cd0034bc0ccb1e9e9266aff 100644
--- a/src/cooling/const_lambda/cooling.h
+++ b/src/cooling/const_lambda/cooling.h
@@ -48,10 +48,9 @@
  * @brief Common operations performed on the cooling function at a
  * given time-step or redshift.
  *
- * @param phys_const The physical constants in internal units.
- * @param us The internal system of units.
  * @param cosmo The current cosmological model.
  * @param cooling The #cooling_function_data used in the run.
+ * @param restart_flag Are we calling this directly after a restart?
  */
 INLINE static void cooling_update(const struct cosmology* cosmo,
                                   struct cooling_function_data* cooling,
diff --git a/src/cooling/grackle/cooling.h b/src/cooling/grackle/cooling.h
index ce71a370656a84c94f5d59c1b3bd75b547234620..03f3f1f14dc008b6ee9feedb09893470a5ccde6f 100644
--- a/src/cooling/grackle/cooling.h
+++ b/src/cooling/grackle/cooling.h
@@ -52,10 +52,9 @@
  * @brief Common operations performed on the cooling function at a
  * given time-step or redshift.
  *
- * @param phys_const The physical constants in internal units.
- * @param us The internal system of units.
  * @param cosmo The current cosmological model.
  * @param cooling The #cooling_function_data used in the run.
+ * @param restart_flag Are we calling this directly after a restart?
  */
 INLINE static void cooling_update(const struct cosmology* cosmo,
                                   struct cooling_function_data* cooling,