Fix not persisting task tests