diff --git a/app/components/sidebar/Menu.client.tsx b/app/components/sidebar/Menu.client.tsx
index 468696b..0472ec4 100644
--- a/app/components/sidebar/Menu.client.tsx
+++ b/app/components/sidebar/Menu.client.tsx
@@ -4,7 +4,6 @@ import { toast } from 'react-toastify';
 import { HistoryItem } from './HistoryItem';
 import { binDates } from './date-binning';
 import { Dialog, DialogButton, DialogDescription, DialogRoot, DialogTitle } from '~/components/ui/Dialog';
-import { IconButton } from '~/components/ui/IconButton';
 import { ThemeSwitch } from '~/components/ui/ThemeSwitch';
 import { db, deleteById, getAll, chatId, type ChatHistoryItem } from '~/lib/persistence';
 import { cubicEasingFn } from '~/utils/easings';
diff --git a/app/routes/api.chat.ts b/app/routes/api.chat.ts
index 78b8abd..cbe6732 100644
--- a/app/routes/api.chat.ts
+++ b/app/routes/api.chat.ts
@@ -35,9 +35,9 @@ async function chatAction({ context, request }: ActionFunctionArgs) {
         const result = await streamText(messages, context.cloudflare.env, options);
 
         return (
-          /*
+          /**
            * WARNING: toAIStream has been removed from streamText.
-           * See migration guide at https://sdk.vercel.ai/docs/migrations
+           * See migration guide at https://sdk.vercel.ai/docs/migrations.
            */
           stream.switchSource(result.toDataStream())
         );
@@ -46,9 +46,9 @@ async function chatAction({ context, request }: ActionFunctionArgs) {
 
     const result = await streamText(messages, context.cloudflare.env, options);
 
-    /*
+    /**
      * WARNING: toAIStream has been removed from streamText.
-     * See migration guide at https://sdk.vercel.ai/docs/migrations
+     * See migration guide at https://sdk.vercel.ai/docs/migrations.
      */
     stream.switchSource(result.toDataStream());
 
diff --git a/app/routes/api.enhancer.ts b/app/routes/api.enhancer.ts
index 25e6e4c..0501b57 100644
--- a/app/routes/api.enhancer.ts
+++ b/app/routes/api.enhancer.ts
@@ -46,12 +46,11 @@ async function enhancerAction({ context, request }: ActionFunctionArgs) {
       },
     });
 
-    const transformedStream =
-      /*
-       * WARNING: toAIStream has been removed from streamText.
-       * See migration guide at https://sdk.vercel.ai/docs/migrations
-       */
-      result.toDataStream().pipeThrough(transformStream);
+    /*
+     * WARNING: toAIStream has been removed from streamText
+     * See migration guide at https://sdk.vercel.ai/docs/migrations
+     */
+    //   result.toDataStream().pipeThrough(transformStream);
 
     return result.toDataStreamResponse();
   } catch (error) {