recompute all token states every step