diff --git a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
index 886494de3c..7321c27769 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
@@ -177,6 +177,7 @@ public void setupSettings() throws IOException {
 
         String jsonEntity = "{\n"
             + "  \"persistent\" : {\n"
+            + "    \"plugins.ml_commons.jvm_heap_memory_threshold\" : 100, \n"
             + "    \"plugins.ml_commons.native_memory_threshold\" : 100 \n"
             + "  }\n"
             + "}";