diff --git a/metadata-ingestion/tests/unit/test_airflow.py b/metadata-ingestion/tests/unit/test_airflow.py
index ef25f1844d..4b310f4216 100644
--- a/metadata-ingestion/tests/unit/test_airflow.py
+++ b/metadata-ingestion/tests/unit/test_airflow.py
@@ -233,7 +233,11 @@ def test_lineage_backend(mock_emit, inlets, outlets):
             )
             op1 >> op2
 
-        ti = TI(task=op2, execution_date=DEFAULT_DATE)
+        if airflow.version.version.startswith("1"):
+            ti = TI(task=op2, execution_date=DEFAULT_DATE)
+        else:
+            ti = TI(task=op2)
+
         ctx1 = {
             "dag": dag,
             "task": op2,