diff --git a/litellm/types/utils.py b/litellm/types/utils.py
index 532162e60f..31e91d7981 100644
--- a/litellm/types/utils.py
+++ b/litellm/types/utils.py
@@ -1518,6 +1518,9 @@ class ImageResponse(OpenAIImageResponse):
         response_ms=None,
         usage: Optional[Usage] = None,
         hidden_params: Optional[dict] = None,
+        id: Optional[str] = None,
+        model: Optional[str] = None,
+        object: Optional[str] = None,
     ):
         if response_ms:
             _response_ms = response_ms
diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx
index 63a4d6f101..310287eb0c 100644
--- a/ui/litellm-dashboard/src/components/networking.tsx
+++ b/ui/litellm-dashboard/src/components/networking.tsx
@@ -69,7 +69,7 @@ const handleError = async (errorData: string) => {
       lastErrorTime = currentTime;
       await sleep(3000); // 5 second sleep
       document.cookie = "token=; expires=Thu, 01 Jan 1970 00:00:00 UTC; path=/;";
-      window.location.href = baseUrl;
+      window.location.href = baseUrl + "ui/";
     }
     lastErrorTime = currentTime;
   } else {
@@ -4653,4 +4653,4 @@ export const teamPermissionsUpdateCall = async (
     console.error("Failed to update team permissions:", error);
     throw error;
   }
-};
\ No newline at end of file
+};