diff --git a/screenshot32/resource/capturescreentest.py b/screenshot32/resource/capturescreentest.py
index e68766fc7797f5dd6e86c9ecad3e70f12ded4ce9..4490bf2395ab2636c4be1d14983910202e897da2 100644
--- a/screenshot32/resource/capturescreentest.py
+++ b/screenshot32/resource/capturescreentest.py
@@ -314,7 +314,7 @@ if __name__ == "__main__":
     if "State=2" not in power_state:
         PrintToLog("SmokeTest:: DISPLAY POWER MANAGER DUMP State=0")
         SysExit()
-    if "distributed on" in args.test_num:
+    if "1/2" in args.test_num or "2/2" in args.test_num:
         EnterShellCmd("param set persist.ace.testmode.enabled 1", 1)
         EnterShellCmd("rm /data/log/hilog/*;hilog -r;hilog -Q pidoff;hilog -Q domainoff;hilog -G 512M;hilog -b D", 1)
         picture_path = os.path.normpath(os.path.join(args.tools_path, "DistributedTest\\testcases"))