diff --git a/integration/airflow/README.md b/integration/airflow/README.md
index 5c4006c7bf..c6f307e0bf 100644
--- a/integration/airflow/README.md
+++ b/integration/airflow/README.md
@@ -243,7 +243,7 @@ t1 = DataProcPySparkOperator(
     job_name=job_name,
     dataproc_pyspark_properties={
         'spark.driver.extraJavaOptions':
-            f"-javaagent:{jar}={os.environ.get('OPENLINEAGE_URL')}/api/v1/namespaces/{os.getenv('OPENLINEAGE_NAMESPACE', 'default')}/jobs/{job_name}/runs/{{{{macros.OpenLineagePlugin.lineage_run_id(task, task_instance)}}}}?api_key={os.environ.get('OPENLINEAGE_API_KEY')}"
+            f"-javaagent:{jar}={os.environ.get('OPENLINEAGE_URL')}/api/v1/namespaces/{os.getenv('OPENLINEAGE_NAMESPACE', 'default')}/jobs/{job_name}/runs/{{{{macros.OpenLineagePlugin.lineage_run_id(task_instance)}}}}?api_key={os.environ.get('OPENLINEAGE_API_KEY')}"
         dag=dag)
 ```
 
diff --git a/integration/airflow/openlineage/airflow/macros.py b/integration/airflow/openlineage/airflow/macros.py
index 0e16b1ab5b..9d16b8e584 100644
--- a/integration/airflow/openlineage/airflow/macros.py
+++ b/integration/airflow/openlineage/airflow/macros.py
@@ -18,7 +18,7 @@ def lineage_run_id(task_instance: "TaskInstance"):
     PythonOperator(
         task_id='render_template',
         python_callable=my_task_function,
-        op_args=['{{ lineage_run_id(task, task_instance) }}'], # lineage_run_id macro invoked
+        op_args=['{{ lineage_run_id(task_instance) }}'], # lineage_run_id macro invoked
         provide_context=False,
         dag=dag
     )
@@ -41,7 +41,7 @@ def lineage_parent_id(task_instance: "TaskInstance"):
     PythonOperator(
         task_id='render_template',
         python_callable=my_task_function,
-        op_args=['{{ lineage_parent_id(task, task_instance) }}'], # macro invoked
+        op_args=['{{ lineage_parent_id(task_instance) }}'], # macro invoked
         provide_context=False,
         dag=dag
     )
diff --git a/integration/airflow/tests/integration/tests/airflow/dags/dbt_dag.py b/integration/airflow/tests/integration/tests/airflow/dags/dbt_dag.py
index 9ea796dff5..3903e9d4df 100644
--- a/integration/airflow/tests/integration/tests/airflow/dags/dbt_dag.py
+++ b/integration/airflow/tests/integration/tests/airflow/dags/dbt_dag.py
@@ -8,7 +8,7 @@
 from airflow.utils.dates import days_ago
 from airflow.version import version as AIRFLOW_VERSION
 
-PLUGIN_MACRO = "{{ macros.OpenLineagePlugin.lineage_parent_id(task, task_instance) }}"
+PLUGIN_MACRO = "{{ macros.OpenLineagePlugin.lineage_parent_id(task_instance) }}"
 
 
 PROJECT_DIR = "/opt/data/dbt/testproject"