Add back tests for tasks controller