diff --git a/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java b/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java
index 78f4163d96d33..8cf7941941bcb 100644
--- a/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java
+++ b/server/src/test/java/org/opensearch/index/query/functionscore/FunctionScoreQueryBuilderTests.java
@@ -948,5 +948,6 @@ public void testVisit() {
         builder.visit(createTestVisitor(visitedQueries));
 
         assertEquals(2, visitedQueries.size());
+        assertTrue(visitedQueries.contains(termQueryBuilder));
     }
 }