Fix a bug that time step was incorrectly updated when multiple time step function exists