diff --git a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/RuleAttributeTrie.java b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/RuleAttributeTrie.java
index 328341998c756..f0199776ccf73 100644
--- a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/RuleAttributeTrie.java
+++ b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/RuleAttributeTrie.java
@@ -19,7 +19,7 @@ public class RuleAttributeTrie implements FastPrefixMatchingStructure {
     private TrieNode root;
 
     /**
-     * Constructs an empty AugmentedTrie.
+     * Constructs an empty Trie.
      */
     public RuleAttributeTrie() {
         root = new TrieNode("");
diff --git a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieDeleter.java b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieDeleter.java
index 3f2dbe9c6e8c3..1a0dc761afdc6 100644
--- a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieDeleter.java
+++ b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieDeleter.java
@@ -9,7 +9,7 @@
 package org.opensearch.plugin.wlm.rule.structure;
 
 /**
- * Handles the deletion operation for the Augmented Trie.
+ * Handles the deletion operation for the Trie.
  */
 class TrieDeleter {
     private TrieNode root;
diff --git a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieInserter.java b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieInserter.java
index 606be603161ba..42ba71c40fe9f 100644
--- a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieInserter.java
+++ b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieInserter.java
@@ -9,7 +9,7 @@
 package org.opensearch.plugin.wlm.rule.structure;
 
 /**
- * Handles the insertion operation for the Augmented Trie.
+ * Handles the insertion operation for the Trie.
  */
 class TrieInserter {
     private TrieNode root;
diff --git a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieNode.java b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieNode.java
index 14b078bd38d7f..a369484016a6e 100644
--- a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieNode.java
+++ b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieNode.java
@@ -16,7 +16,7 @@
 import java.util.Queue;
 
 /**
- * Represents a node in the Augmented Trie.
+ * Represents a node in the Trie.
  * Each node contains a key, an optional value, and references to child nodes.
  */
 class TrieNode {
diff --git a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieSearcher.java b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieSearcher.java
index 0dd21bad4a3d5..279837b80dfcc 100644
--- a/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieSearcher.java
+++ b/plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/rule/structure/TrieSearcher.java
@@ -14,7 +14,7 @@
 import java.util.function.Function;
 
 /**
- * Handles the search operation for the Augmented Trie.
+ * Handles the search operation for the Trie.
  */
 class TrieSearcher {
     private TrieNode root;