You can use `masked_token.first(LENGTH)` here.