Forgot to import task decorator