2 files changed, 2 insertions, 2 deletions
diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts
index 62851019..2728c394 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.ts
+++ b/packages/cli/src/ui/hooks/useGeminiStream.ts
@@ -30,11 +30,11 @@ import {
   type Part,
 } from '@google/genai';
 import {
+  StreamingState,
   HistoryItem,
   IndividualToolCallDisplay,
   ToolCallStatus,
 } from '../types.js';
-import { StreamingState } from '../../core/gemini-stream.js';
 import { toolRegistry } from '../../tools/tool-registry.js';
 
 const addHistoryItem = (
diff --git a/packages/cli/src/ui/hooks/useLoadingIndicator.ts b/packages/cli/src/ui/hooks/useLoadingIndicator.ts
index 1dab9756..6d1d77d4 100644
--- a/packages/cli/src/ui/hooks/useLoadingIndicator.ts
+++ b/packages/cli/src/ui/hooks/useLoadingIndicator.ts
@@ -9,7 +9,7 @@ import {
   WITTY_LOADING_PHRASES,
   PHRASE_CHANGE_INTERVAL_MS,
 } from '../constants.js';
-import { StreamingState } from '../../core/gemini-stream.js';
+import { StreamingState } from '../types.js';
 
 export const useLoadingIndicator = (streamingState: StreamingState) => {
   const [elapsedTime, setElapsedTime] = useState(0);