Update tests to reflect changes JSON changes.