diff --git a/packages/vscode-ide-companion/src/webview/components/messages/AssistantMessage.tsx b/packages/vscode-ide-companion/src/webview/components/messages/AssistantMessage.tsx
index b1f2582c..666c6b0b 100644
--- a/packages/vscode-ide-companion/src/webview/components/messages/AssistantMessage.tsx
+++ b/packages/vscode-ide-companion/src/webview/components/messages/AssistantMessage.tsx
@@ -7,7 +7,6 @@
 import type React from 'react';
 import { MessageContent } from '../MessageContent.js';
 import './AssistantMessage.css';
-import './SimpleTimeline.css';
 
 interface AssistantMessageProps {
   content: string;
@@ -56,7 +55,7 @@ export const AssistantMessage: React.FC<AssistantMessageProps> = ({
 
   return (
     <div
-      className={`qwen-message message-item assistant-message-container simple-timeline-container ${getStatusClass()}`}
+      className={`qwen-message message-item assistant-message-container ${getStatusClass()}`}
       style={{
         width: '100%',
         alignItems: 'flex-start',
diff --git a/packages/vscode-ide-companion/src/webview/components/toolcalls/shared/LayoutComponents.tsx b/packages/vscode-ide-companion/src/webview/components/toolcalls/shared/LayoutComponents.tsx
index b77583ab..8d0395cd 100644
--- a/packages/vscode-ide-companion/src/webview/components/toolcalls/shared/LayoutComponents.tsx
+++ b/packages/vscode-ide-companion/src/webview/components/toolcalls/shared/LayoutComponents.tsx
@@ -10,7 +10,6 @@
 import type React from 'react';
 import { FileLink } from '../../ui/FileLink.js';
 import './LayoutComponents.css';
-import './SimpleTimeline.css';
 
 /**
  * Props for ToolCallContainer - Claude Code style layout