diff --git a/packages/cli/src/ui/commands/compressCommand.test.ts b/packages/cli/src/ui/commands/compressCommand.test.ts
index 41d0ff5b09..a0c9418817 100644
--- a/packages/cli/src/ui/commands/compressCommand.test.ts
+++ b/packages/cli/src/ui/commands/compressCommand.test.ts
@@ -22,10 +22,6 @@ describe('compressCommand', () => {
   let context: ReturnType<typeof createMockCommandContext>;
   let mockTryCompressChat: ReturnType<typeof vi.fn>;
 
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
   beforeEach(() => {
     mockTryCompressChat = vi.fn();
     vi.mocked(Core.tokenLimit).mockReturnValue(1000);
diff --git a/packages/cli/src/ui/components/messages/CompressionMessage.tsx b/packages/cli/src/ui/components/messages/CompressionMessage.tsx
index 3cf3d33c2d..2d17c4b6e9 100644
--- a/packages/cli/src/ui/components/messages/CompressionMessage.tsx
+++ b/packages/cli/src/ui/components/messages/CompressionMessage.tsx
@@ -32,7 +32,7 @@ export function CompressionMessage({
 
     switch (compressionStatus) {
       case CompressionStatus.COMPRESSED:
-        return `Context compressed (${beforePercentage}% → ${afterPercentage}%).`;
+        return `Context compressed (${beforePercentage}% ➔ ${afterPercentage}%).`;
       case CompressionStatus.COMPRESSION_FAILED_INFLATED_TOKEN_COUNT:
         return 'Compression was not beneficial for this history size.';
       case CompressionStatus.COMPRESSION_FAILED_TOKEN_COUNT_ERROR:
diff --git a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
index 1e3a6a2b89..fb2fca9108 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
+++ b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
@@ -256,12 +256,9 @@ describe('useGeminiStream', () => {
   const mockOnAuthError = vi.fn();
   const mockPerformMemoryRefresh = vi.fn(() => Promise.resolve());
   const mockSetModelSwitchedFromQuotaError = vi.fn();
+  const mockOnCancelSubmit = vi.fn();
   const mockSetShellInputFocused = vi.fn();
 
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
   const mockGetGeminiClient = vi.fn().mockImplementation(() => {
     const clientInstance = new MockedGeminiClientClass(mockConfig);
     return clientInstance;
@@ -339,10 +336,6 @@ describe('useGeminiStream', () => {
     getContextWindowCompressionThreshold: vi.fn(() => 0.2),
   } as unknown as Config;
 
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
   beforeEach(() => {
     vi.clearAllMocks(); // Clear mocks before each test
     mockAddItem = vi.fn();