Add missing task classes to directives