small fix for removed Token import