With the recent changes in the output format as List, updated tests as well so they validate on the new format of output.