Updates to tests for tasks functionality