Put token class directly on the directive
Avoid color flickering in dependencyTree