You can use masked_token.last(LENGTH) here.