Fix to timestep functionality.