diff --git a/tests/benchmark/milvus_benchmark/update.py b/tests/benchmark/milvus_benchmark/update.py
index e0b90c5875..3b479abc35 100644
--- a/tests/benchmark/milvus_benchmark/update.py
+++ b/tests/benchmark/milvus_benchmark/update.py
@@ -75,7 +75,7 @@ def update_values(src_values_file, deploy_params_file):
         cpus = res["cpus"]
         mems = res["mems"]
         gpus = res["gpus"]
-    if cpus:
+    if cpus and mems:
         # Set the scope of cpu application according to the configuration
         resources = {
                 "limits": {