Fix task imports