Yeah, I think `math.floor` is redundant. `time/self._time_step` will work.