diff --git a/lang/include/ivy/lang/codegen.h b/lang/include/ivy/lang/codegen.h
index 81d570c..7a4c5fc 100644
--- a/lang/include/ivy/lang/codegen.h
+++ b/lang/include/ivy/lang/codegen.h
@@ -10,8 +10,10 @@ struct ivy_ast_node;
 enum ivy_ast_iteration_type;
 
 struct mie_module;
+struct mie_ctx;
 
-IVY_API enum ivy_status ivy_codegen_create(struct ivy_codegen **out);
+IVY_API enum ivy_status ivy_codegen_create(
+	struct mie_ctx *ctx, struct ivy_codegen **out);
 IVY_API void ivy_codegen_destroy(struct ivy_codegen *gen);
 
 IVY_API enum ivy_status ivy_codegen_start_module(struct ivy_codegen *gen);