diff --git a/kitty_tests/graphics.py b/kitty_tests/graphics.py
index 6560f8df7..fc268afde 100644
--- a/kitty_tests/graphics.py
+++ b/kitty_tests/graphics.py
@@ -211,6 +211,8 @@ class TestGraphics(BaseTest):
         data.clear()
         for i in range(25):
             self.assertIsNone(add(i, f'{i}' * i))
+        dc.wait_for_write()
+        check_data()
 
         before = dc.size_on_disk()
         while dc.total_size > before // 3: