Fixed Test Cases for TaskBuilder