Fixes token update bug.