Use API data for lineage UI testing; small modifications