diff --git a/tests/test_main.py b/tests/test_main.py
index b325c53..a1eed5f 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,11 +1,11 @@
 from pathlib import Path
 
-from autora.doc.pipelines.main import eval, evaluate_documentation, generate, import_data
-from autora.doc.runtime.prompts import InstructionPrompts, SystemPrompts
-
 import jsonlines
 import pytest
 
+from autora.doc.pipelines.main import eval, evaluate_documentation, generate, import_data
+from autora.doc.runtime.prompts import InstructionPrompts, SystemPrompts
+
 # dummy HF model for testing
 TEST_HF_MODEL = "hf-internal-testing/tiny-random-FalconForCausalLM"