diff --git a/core/src/types/message/messageEntity.ts b/core/src/types/message/messageEntity.ts
index 26bcad1a7..b74176fd4 100644
--- a/core/src/types/message/messageEntity.ts
+++ b/core/src/types/message/messageEntity.ts
@@ -97,7 +97,7 @@ export enum ErrorCode {
  */
 export enum ContentType {
   Text = 'text',
-  Image = 'image',
+  Image = 'image_url',
   Pdf = 'pdf',
 }
 
diff --git a/web/containers/ErrorMessage/index.tsx b/web/containers/ErrorMessage/index.tsx
index b572da636..2af6065a6 100644
--- a/web/containers/ErrorMessage/index.tsx
+++ b/web/containers/ErrorMessage/index.tsx
@@ -46,7 +46,7 @@ const ErrorMessage = ({ message }: { message: ThreadMessage }) => {
   }
 
   const getEngine = () => {
-    const engineName = activeThread?.assistants?.[0]?.model?.engine
+    const engineName = activeAssistant?.model?.engine
     return engineName ? EngineManager.instance().get(engineName) : null
   }