diff --git a/kitty_tests/choose.py b/kitty_tests/choose.py
index 5dee551fc..623a8ca08 100644
--- a/kitty_tests/choose.py
+++ b/kitty_tests/choose.py
@@ -11,7 +11,7 @@ from . import BaseTest
 def run(input_data, query, **kw):
     kw['threads'] = kw.get('threads', 1)
     mark = kw.pop('mark', False)
-    from kittens.choose.main import match
+    from kittens.choose.match import match
     mark_before = mark_after = ''
     if mark:
         if mark is True: