diff --git a/merico/unit_tests/main.py b/merico/unit_tests/main.py
index ec4952f..31f796a 100644
--- a/merico/unit_tests/main.py
+++ b/merico/unit_tests/main.py
@@ -3,10 +3,10 @@ import os
 import sys
 
 import click
-from cache import LocalCache
-
 import openai
 
+from cache import LocalCache
+
 sys.path.append(os.path.dirname(__file__))
 
 from i18n import TUILanguage, get_translation
@@ -17,7 +17,6 @@ from model import (
 )
 from ut_workflow import UnitTestsWorkflow
 
-
 from lib.chatmark import Step
 from lib.ide_service import IDEService
 
diff --git a/merico/unit_tests/ut_workflow.py b/merico/unit_tests/ut_workflow.py
index d7e72fe..052ecbe 100644
--- a/merico/unit_tests/ut_workflow.py
+++ b/merico/unit_tests/ut_workflow.py
@@ -1,4 +1,3 @@
-from cache import LocalCache
 from typing import Dict, List, Tuple
 
 from find_context import (
@@ -18,6 +17,7 @@ from propose_test import propose_test
 from tools.file_util import retrieve_file_content
 from write_tests import write_and_print_tests
 
+from cache import LocalCache
 from lib.chatmark import Checkbox, Form, Step, TextEditor