diff --git a/diag/ctx.c b/diag/ctx.c
index 651e61b..39a7152 100644
--- a/diag/ctx.c
+++ b/diag/ctx.c
@@ -24,7 +24,7 @@ void ivy_diag_ctx_destroy(struct ivy_diag_ctx *ctx)
 {
 }
 
-enum ivy_status ivy_diag_set_line_source(
+enum ivy_status ivy_diag_ctx_set_line_source(
 	struct ivy_diag_ctx *ctx, struct ivy_line_source *src)
 {
 	ctx->ctx_line_source = src;
diff --git a/diag/include/ivy/diag.h b/diag/include/ivy/diag.h
index 4ab880b..5429c59 100644
--- a/diag/include/ivy/diag.h
+++ b/diag/include/ivy/diag.h
@@ -141,7 +141,7 @@ extern void ivy_diag_stream_init_tty(
 extern enum ivy_status ivy_diag_ctx_create(struct ivy_diag_ctx **out);
 extern void ivy_diag_ctx_destroy(struct ivy_diag_ctx *ctx);
 
-extern enum ivy_status ivy_diag_set_line_source(
+extern enum ivy_status ivy_diag_ctx_set_line_source(
 	struct ivy_diag_ctx *ctx, struct ivy_line_source *src);
 extern enum ivy_status ivy_diag_ctx_set_class_definitions(
 	struct ivy_diag_ctx *ctx, const struct ivy_diag_class *classes,
diff --git a/test/diag.c b/test/diag.c
index 0299b50..5653662 100644
--- a/test/diag.c
+++ b/test/diag.c
@@ -84,7 +84,7 @@ int main(void)
 		return -1;
 	}
 
-	ivy_diag_set_line_source(ctx, &source_file);
+	ivy_diag_ctx_set_line_source(ctx, &source_file);
 	ivy_diag_ctx_set_class_definitions(ctx, diag_classes, nr_diag_classes);
 	ivy_diag_ctx_set_msg_definitions(ctx, diag_msg, nr_diag_msg);