Small fix of big bug

Fix token generate