diff --git a/src/internal/apimagic_driver_impl.h b/src/internal/apimagic_driver_impl.h
index 6d5ce02..5ddf234 100644
--- a/src/internal/apimagic_driver_impl.h
+++ b/src/internal/apimagic_driver_impl.h
@@ -64,4 +64,16 @@ static inline struct amgc_driver_ctx_impl * amgc_get_driver_ictx(
         return 0;
 }
 
+static inline void amgc_driver_set_ectx(
+	const struct amgc_driver_ctx * dctx,
+	const struct amgc_unit_ctx *   uctx,
+	const char *                   unit)
+{
+	struct amgc_driver_ctx_impl *  ictx;
+
+	ictx        = amgc_get_driver_ictx(dctx);
+	ictx->euctx = uctx;
+	ictx->eunit = unit;
+}
+
 #endif