diff --git a/kitty_tests/layout.py b/kitty_tests/layout.py
index 9488c3cb1..8e47fdf9b 100644
--- a/kitty_tests/layout.py
+++ b/kitty_tests/layout.py
@@ -2,10 +2,11 @@
 # vim:fileencoding=utf-8
 # License: GPL v3 Copyright: 2018, Kovid Goyal <kovid at kovidgoyal.net>
 
-from . import BaseTest
 from kitty.config import defaults
-from kitty.layout import Stack, Horizontal, idx_for_id
 from kitty.fast_data_types import pt_to_px
+from kitty.layout import Horizontal, Stack, Tall, Grid, idx_for_id
+
+from . import BaseTest
 
 
 class Window:
@@ -172,11 +173,11 @@ class TestLayout(BaseTest):
         check_visible()
 
     def test_layout_operations(self):
-        for layout_class in Stack, Horizontal:
+        for layout_class in (Stack, Horizontal, Tall, Grid):
             q = create_layout(layout_class)
             self.do_ops_test(q)
 
     def test_overlay_layout_operations(self):
-        for layout_class in Stack, Horizontal:
+        for layout_class in (Stack, Horizontal, Tall, Grid):
             q = create_layout(layout_class)
             self.do_overlay_test(q)