Remove leftover reference to Task class