Make timestep updating work for both models.