Fix TokenMap test