diff --git a/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnector.java b/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnector.java
index 00fa2004b32..d7cfff53e9e 100644
--- a/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnector.java
+++ b/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnector.java
@@ -214,7 +214,7 @@ public void testCreateCatalog() throws Exception {
     assertThat(computeActual("show catalogs").getOnlyColumnAsSet()).contains("memory");
 
     // testing the gravitino connector framework works.
-    assertThat(computeActual("select * from system.jdbc.tables"));
+    assertThat(computeActual("select * from system.jdbc.tables").getRowCount()).isGreaterThan(1);
 
     // test metalake named test. the connector name is gravitino
     assertUpdate("call gravitino.system.create_catalog('memory1', 'memory', Map())");
diff --git a/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnectorWithMetalakeCatalogName.java b/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnectorWithMetalakeCatalogName.java
index a7691c65cd6..0ca3749f006 100644
--- a/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnectorWithMetalakeCatalogName.java
+++ b/trino-connector/src/test/java/com/datastrato/gravitino/trino/connector/TestGravitinoConnectorWithMetalakeCatalogName.java
@@ -95,7 +95,7 @@ public void testCreateCatalog() throws Exception {
     assertThat(computeActual("show catalogs").getOnlyColumnAsSet()).contains("test.memory");
 
     // testing the gravitino connector framework works.
-    assertThat(computeActual("select * from system.jdbc.tables"));
+    assertThat(computeActual("select * from system.jdbc.tables").getRowCount()).isGreaterThan(1);
 
     // test metalake named test. the connector name is gravitino
     assertUpdate("call gravitino.system.create_catalog('memory1', 'memory', Map())");