Updated import in task