Fixes token offset registration.