@@ -217,7 +217,7 @@ def test_feature_get_online_features_types_match(online_types_test_fixtures):
     )
     fs = environment.feature_store
     features = [fv.name + ":value"]
-    entity = driver(value_type=ValueType.UNKNOWN)
+    entity = driver(value_type=config.entity_type)
     fs.apply([fv, entity])
     fs.materialize(environment.start_date, environment.end_date)
Original file line number	Diff line number	Diff line change
`@@ -217,7 +217,7 @@ def test_feature_get_online_features_types_match(online_types_test_fixtures):`
`217`	`217`	`)`
`218`	`218`	`fs = environment.feature_store`
`219`	`219`	`features = [fv.name + ":value"]`
`220`		`- entity = driver(value_type=ValueType.UNKNOWN)`
	`220`	`+ entity = driver(value_type=config.entity_type)`
`221`	`221`	`fs.apply([fv, entity])`
`222`	`222`	`fs.materialize(environment.start_date, environment.end_date)`
`223`	`223`