Fix assignTokens()