diff --git a/src/lib/components/chat/Chat.svelte b/src/lib/components/chat/Chat.svelte
index ae584ba8f..0ef11313e 100644
--- a/src/lib/components/chat/Chat.svelte
+++ b/src/lib/components/chat/Chat.svelte
@@ -216,7 +216,7 @@
 				} else {
 					message.statusHistory = [data];
 				}
-			} else if (type === 'citation') {
+			} else if (type === 'source') {
 				if (data?.type === 'code_execution') {
 					// Code execution; update existing code execution by ID, or add new one.
 					if (!message?.code_executions) {
@@ -235,7 +235,7 @@
 
 					message.code_executions = message.code_executions;
 				} else {
-					// Regular citation.
+					// Regular source.
 					if (message?.sources) {
 						message.sources.push(data);
 					} else {