diff --git a/esmf_regrid/__init__.py b/esmf_regrid/__init__.py
index 9c6070d6..2472b508 100644
--- a/esmf_regrid/__init__.py
+++ b/esmf_regrid/__init__.py
@@ -16,7 +16,7 @@
         raise exc
 
 if hasattr(_imesh, "PARSE_UGRID_ON_LOAD"):
-    _load_context = _imesh.PARSE_UGRID_ON_LOAD
+    _load_context = _imesh.PARSE_UGRID_ON_LOAD.context
 else:
     from contextlib import nullcontext