diff --git a/tests/ml_commons/test_ml_commons_client.py b/tests/ml_commons/test_ml_commons_client.py
index b7ad37ac..b8df8b4a 100644
--- a/tests/ml_commons/test_ml_commons_client.py
+++ b/tests/ml_commons/test_ml_commons_client.py
@@ -583,7 +583,7 @@ def test_stats():
     assert len(res["nodes"]) > 0
 
     node_id = list(res["nodes"].keys())[0]
-    stat_id = "ml_jvm_heap_usage"
+    stat_id = "ml_executing_task_count"
 
     res = ml_client.get_stats(node_id, stat_id)