diff --git a/ChatQnA/ui/svelte/src/lib/shared/components/loading/Loading.svelte b/ChatQnA/ui/svelte/src/lib/shared/components/loading/Loading.svelte
index 295ff1da8..51e89cfe7 100644
--- a/ChatQnA/ui/svelte/src/lib/shared/components/loading/Loading.svelte
+++ b/ChatQnA/ui/svelte/src/lib/shared/components/loading/Loading.svelte
@@ -15,7 +15,7 @@
 -->
 
 <div
-	class="flex items-center justify-center self-center bg-black text-sm text-gray-500"
+	class="mb-6 flex items-center justify-center self-center bg-black text-sm text-gray-500"
 />
 <div class="flex items-center justify-center gap-3">
 	<div class="relative inline-flex">
diff --git a/ChatQnA/ui/svelte/src/lib/shared/constant/Interface.ts b/ChatQnA/ui/svelte/src/lib/shared/constant/Interface.ts
index c461e2b79..221f17a26 100644
--- a/ChatQnA/ui/svelte/src/lib/shared/constant/Interface.ts
+++ b/ChatQnA/ui/svelte/src/lib/shared/constant/Interface.ts
@@ -35,7 +35,6 @@ type Map<T> = T extends MessageType.Text | MessageType.SingleAudio
 			: { imgSrc: string; imgId: string }[];
 
 export interface Message {
-	imgSrc: string | null | undefined;
 	role: MessageRole;
 	type: MessageType;
 	content: Map<Message["type"]>;
diff --git a/VisualQnA/ui/svelte/src/lib/shared/constant/Interface.ts b/VisualQnA/ui/svelte/src/lib/shared/constant/Interface.ts
index 221f17a26..c461e2b79 100644
--- a/VisualQnA/ui/svelte/src/lib/shared/constant/Interface.ts
+++ b/VisualQnA/ui/svelte/src/lib/shared/constant/Interface.ts
@@ -35,6 +35,7 @@ type Map<T> = T extends MessageType.Text | MessageType.SingleAudio
 			: { imgSrc: string; imgId: string }[];
 
 export interface Message {
+	imgSrc: string | null | undefined;
 	role: MessageRole;
 	type: MessageType;
 	content: Map<Message["type"]>;