Attempt at fixing weirdness in time_heat