diff --git a/tests/search/custom_state_api/custom_state_api.rb b/tests/search/custom_state_api/custom_state_api.rb
index d9709cde1..863e85c12 100644
--- a/tests/search/custom_state_api/custom_state_api.rb
+++ b/tests/search/custom_state_api/custom_state_api.rb
@@ -99,7 +99,7 @@ def assert_flush_engine(page)
   end
 
   def assert_tls(page)
-    assert_keys(["test"], page)
+    assert_keys(["size_on_disk", "test"], page)
   end
 
   def assert_hw_info(page)