diff --git a/script/run-mlperf-inference-app/customize.py b/script/run-mlperf-inference-app/customize.py
index 5c3932a3d..3d1886e1a 100644
--- a/script/run-mlperf-inference-app/customize.py
+++ b/script/run-mlperf-inference-app/customize.py
@@ -305,7 +305,7 @@ def preprocess(i):
             if state.get('docker', {}):
                 del (state['docker'])
 
-        if env.get("MLC_MLPERF_LOADGEN_COMPLIANCE", "") == "yes":
+        if is_true(env.get("MLC_MLPERF_LOADGEN_COMPLIANCE", "")):
             for test in test_list:
                 env['MLC_MLPERF_LOADGEN_COMPLIANCE_TEST'] = test
                 env['MLC_MLPERF_LOADGEN_MODE'] = "compliance"