diff --git a/src/dbt_osmosis/vendored/dbt_core_interface/project.py b/src/dbt_osmosis/vendored/dbt_core_interface/project.py
index 8702af6..e914a92 100644
--- a/src/dbt_osmosis/vendored/dbt_core_interface/project.py
+++ b/src/dbt_osmosis/vendored/dbt_core_interface/project.py
@@ -87,7 +87,8 @@
 
 try:
     # dbt >= 1.8
-    from dbt_common.clients.system import make_directory
+    from dbt_common.clients.system import make_directory, get_env
+    from dbt_common.context import set_invocation_context
 except ImportError:
     # dbt < 1.8
     from dbt.clients.system import make_directory
@@ -386,6 +387,8 @@ def __init__(
             project_dir=project_dir or DEFAULT_PROJECT_DIR,
             profile=profile,
         )
+        if hasattr(sys.modules[__name__], 'set_invocation_context'):
+            set_invocation_context(get_env())
         if vars is None:
             vars = "{}"
         self.base_config.vars = vars