Add run_on_start to task