diff --git a/vite/src/ChatMessage.svelte b/vite/src/ChatMessage.svelte
index 4a777c7..a917232 100644
--- a/vite/src/ChatMessage.svelte
+++ b/vite/src/ChatMessage.svelte
@@ -120,11 +120,11 @@
         padding: 16px;
 
         &.assistant {
-            background-color:  var(--bs-gray-300);
+            background-color:  var(--bs-primary-bg-subtle);
         }
 
         &.user {
-            background-color: var(--bs-gray-500);
+            background-color: var(--bs-warning-bg-subtle);
         }
     }
 </style>