diff --git a/test/test_generate_branch_name.py b/test/test_generate_branch_name.py
index d660756..221365b 100644
--- a/test/test_generate_branch_name.py
+++ b/test/test_generate_branch_name.py
@@ -1,4 +1,4 @@
-from aider_gitea.__main__ import generate_branch_name
+from aider_gitea import generate_branch_name
 
 
 def test_generate_branch_name_normal():