diff --git a/kitty_tests/screen.py b/kitty_tests/screen.py
index 83db04d47..c682c05ac 100644
--- a/kitty_tests/screen.py
+++ b/kitty_tests/screen.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>
 
-from kitty.fast_data_types import DECAWM, DECCOLM, DECOM, IRM, VT_PARSER_BUFFER_SIZE, Cursor, ESC_CSI
+from kitty.fast_data_types import DECAWM, DECCOLM, DECOM, IRM, VT_PARSER_BUFFER_SIZE, Cursor
 from kitty.marks import marker_from_function, marker_from_regex
 from kitty.window import pagerhist