improve test for Task