diff --git a/tests/integration/query_execute_test.go b/tests/integration/query_execute_test.go
index 56aed60c2..51308e005 100644
--- a/tests/integration/query_execute_test.go
+++ b/tests/integration/query_execute_test.go
@@ -113,7 +113,7 @@ func TestQueryExecute(t *testing.T) {
 			_, has := plan[key]
 			require.True(t, has, key)
 		}
-		require.NotEmpty(t, ast)
+		require.Contains(t, ast, "return")
 	})
 	t.Run("Scan", func(t *testing.T) {
 		var (