diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py
index 00342319c52e4..52d754b3132e2 100644
--- a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py
+++ b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_dag_events_false_dag.py
@@ -35,7 +35,7 @@
 
 
 def _register_hook_lineage():
-    from airflow.lineage.hook import get_hook_lineage_collector
+    from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector
 
     collector = get_hook_lineage_collector()
     collector.add_input_asset(context=None, uri="file://host1/in1.txt")
diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py
index 721672808b85d..6f1e8a76f972f 100644
--- a/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py
+++ b/providers/openlineage/tests/system/openlineage/example_openlineage_policy_hook_lineage_dag.py
@@ -37,7 +37,7 @@
 
 
 def _register_hook_lineage():
-    from airflow.lineage.hook import get_hook_lineage_collector
+    from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector
 
     collector = get_hook_lineage_collector()
     collector.add_input_asset(context=None, uri="file://host1/in1.txt")
@@ -47,7 +47,7 @@ def _register_hook_lineage():
 
 
 def _register_sql_hook_lineage():
-    from airflow.lineage.hook import get_hook_lineage_collector
+    from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector
     from airflow.providers.common.sql.hooks.lineage import SqlJobHookLineageExtra
 
     collector = get_hook_lineage_collector()