diff --git a/aider/io.py b/aider/io.py
index 04bd9f64b..5e27a267a 100644
--- a/aider/io.py
+++ b/aider/io.py
@@ -3,6 +3,7 @@ import os
 from collections import defaultdict
 from dataclasses import dataclass
 from datetime import datetime
+from io import StringIO
 from pathlib import Path
 
 from prompt_toolkit.completion import Completer, Completion, ThreadedCompleter
@@ -714,9 +715,6 @@ class InputOutput:
 
             return "\n".join(read_only_files + editable_files) + "\n"
 
-        # Use rich Columns for pretty output
-        from io import StringIO
-
         output = StringIO()
         console = Console(file=output, force_terminal=False)