Revert "Minor changes in tests for TaskBuilderWithToken."

This reverts commit 81e726ad8da45b0490f14bcd74af25421bd0b88c.