@@ -38,7 +38,7 @@ def test_online_retrieval(environment, universal_data_sources, benchmark):
     sample_customers = random.sample(entities.customer_vals, 10)
 
     entity_rows = [
-        {"driver": d, "customer_id": c, "val_to_add": 50}
+        {"driver_id": d, "customer_id": c, "val_to_add": 50}
         for (d, c) in zip(sample_drivers, sample_customers)
     ]
Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ def test_online_retrieval(environment, universal_data_sources, benchmark):`
`38`	`38`	`sample_customers = random.sample(entities.customer_vals, 10)`
`39`	`39`
`40`	`40`	`entity_rows = [`
`41`		`- {"driver": d, "customer_id": c, "val_to_add": 50}`
	`41`	`+ {"driver_id": d, "customer_id": c, "val_to_add": 50}`
`42`	`42`	`for (d, c) in zip(sample_drivers, sample_customers)`
`43`	`43`	`]`
`44`	`44`