fix token embedding