Address Martin Comments

Signed-off-by: Varun Jain <[email protected]>

src/main/java/org/opensearch/neuralsearch/search/query/TopDocsMerger.java

@@ -78,7 +78,7 @@ private TotalHits getMergedTotalHits(final TopDocsAndMaxScore source, final TopD
     }
 
     private TopDocs getTopDocs(ScoreDoc[] mergedScoreDocs, TotalHits mergedTotalHits) {
-        if (sortAndFormats != null) {
+        if (isSortingEnabled()) {
             return new TopFieldDocs(mergedTotalHits, mergedScoreDocs, sortAndFormats.sort.getSort());
         }
         return new TopDocs(mergedTotalHits, mergedScoreDocs);