diff --git a/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/FlowModelGeneratorService.java b/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/FlowModelGeneratorService.java
index 7a1301097..8aab9e0c7 100644
--- a/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/FlowModelGeneratorService.java
+++ b/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/FlowModelGeneratorService.java
@@ -504,7 +504,7 @@ public CompletableFuture<FlowModelAvailableNodesResponse> search(SearchRequest r
                 }
 
                 SearchCommand.Kind searchKind = SearchCommand.Kind.valueOf(request.searchKind());
-                SearchCommand command = SearchCommand.from(searchKind, module, request.position(), request.query());
+                SearchCommand command = SearchCommand.from(searchKind, module, request.position(), request.queryMap());
                 response.setCategories(command.execute());
             } catch (Throwable e) {
                 response.setError(e);
diff --git a/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/request/SearchRequest.java b/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/request/SearchRequest.java
index 1f9dc5d5e..ff6e90072 100644
--- a/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/request/SearchRequest.java
+++ b/flow-model-generator/modules/flow-model-generator-ls-extension/src/main/java/io/ballerina/flowmodelgenerator/extension/request/SearchRequest.java
@@ -28,9 +28,9 @@
  * @param searchKind the kind of search to perform (corresponds to SearchCommand.Kind)
  * @param filePath the path of the file (may be null for certain search kinds)
  * @param position the line range of the current position (may be null for certain search kinds)
- * @param query the map containing query parameters
+ * @param queryMap the map containing queryMap parameters
  * @since 2.0.0
  */
-public record SearchRequest(String searchKind, String filePath, LineRange position, Map<String, String> query) {
+public record SearchRequest(String searchKind, String filePath, LineRange position, Map<String, String> queryMap) {
 
 }