Fix taskchain test

Incorporates your commit and my suggested change.