Fix bug in test, task completed