diff --git a/application/storage/local.py b/application/storage/local.py
index cf0ebcb8..a987b32b 100644
--- a/application/storage/local.py
+++ b/application/storage/local.py
@@ -26,7 +26,7 @@ class LocalStorage(BaseStorage):
             return path
         return os.path.join(self.base_dir, path)
 
-    def save_file(self, file_data: BinaryIO, path: str) -> dict:
+    def save_file(self, file_data: BinaryIO, path: str, **kwargs) -> dict:
         """Save a file to local storage."""
         full_path = self._get_full_path(path)
 
diff --git a/frontend/src/components/AgentImage.tsx b/frontend/src/components/AgentImage.tsx
index 3c7ab25b..28109b49 100644
--- a/frontend/src/components/AgentImage.tsx
+++ b/frontend/src/components/AgentImage.tsx
@@ -1,4 +1,4 @@
-import { useState } from 'react';
+import { useState, useEffect } from 'react';
 import Robot from '../assets/robot.svg';
 
 type AgentImageProps = {
@@ -14,8 +14,16 @@ export default function AgentImage({
   className = '',
   fallbackSrc = Robot,
 }: AgentImageProps) {
-  const initialSrc = src && src.trim() !== '' ? src : fallbackSrc;
-  const [currentSrc, setCurrentSrc] = useState(initialSrc);
+  const [currentSrc, setCurrentSrc] = useState(
+    src && src.trim() !== '' ? src : fallbackSrc,
+  );
+
+  useEffect(() => {
+    const newSrc = src && src.trim() !== '' ? src : fallbackSrc;
+    if (newSrc !== currentSrc) {
+      setCurrentSrc(newSrc);
+    }
+  }, [src, fallbackSrc]);
 
   return (
     <img
@@ -30,5 +38,3 @@ export default function AgentImage({
     />
   );
 }
-
-