fix the time stepping heuristic

bd3c075bbff5bf280f5b7fa5a19ee3ff314079f8 was only for the current
episode, and it did also reduce the step size if the episode was over
after the current time step...