Add missing task import