improve token setter