Fix: detect special tokens