I don't think you need to do this - the sub_span_after_token will only give you the span for one token