diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py
index c672c27cb..7db2faaea 100644
--- a/cosmos/operators/local.py
+++ b/cosmos/operators/local.py
@@ -46,7 +46,11 @@
 if TYPE_CHECKING:
     from airflow.datasets import Dataset  # noqa: F811
     from dbt.cli.main import dbtRunner, dbtRunnerResult
-    from openlineage.client.run import RunEvent
+
+    try:  # pragma: no cover
+        from openlineage.client.event_v2 import RunEvent  # pragma: no cover
+    except ImportError:  # pragma: no cover
+        from openlineage.client.run import RunEvent  # pragma: no cover
 
 
 from sqlalchemy.orm import Session