diff --git a/kitty_tests/screen.py b/kitty_tests/screen.py
index 09adbe5dd..24086cf0c 100644
--- a/kitty_tests/screen.py
+++ b/kitty_tests/screen.py
@@ -1100,9 +1100,20 @@ class TestScreen(BaseTest):
         self.ae(tuple(map(int, lco().split())), tuple(range(0, 30)))
 
         # last non empty command output
+        s = self.create_screen()
         draw_prompt('a'), draw_output(2, 'a')
         draw_prompt('b'), mark_output()
         self.ae(lco(), '')
         self.ae(lco(which=3), '0a\n1a')
         s.draw('running'), s.index(),  s.carriage_return()
         self.ae(lco(which=3), 'running\n')
+        s = self.create_screen()
+        draw_prompt('p1')
+        draw_output(30)
+        self.ae(tuple(map(int, lco(which=3).split())), tuple(range(0, 30)))
+        s = self.create_screen()
+        draw_prompt('p1')
+        draw_output(2, 'a')
+        draw_prompt('p1')
+        draw_prompt('p1')
+        self.ae(lco(which=3), '0a\n1a')