diff --git a/src/core/command/run.py b/src/core/command/run.py
index c159190fdacd824ca00e3a5582ef6dcb87a74dfd..aefea4ca060bf18a68026e01e1ac73facf5b77ba 100644
--- a/src/core/command/run.py
+++ b/src/core/command/run.py
@@ -233,7 +233,6 @@ class Run(object):
 
             scheduler = get_plugin(plugin_type=Plugin.SCHEDULER,
                                    plugin_id=SchedulerType.SCHEDULER)[0]
-            scheduler.update_test_type_in_source("OHRust", DeviceTestType.oh_rust_test)
             if scheduler is None:
                 LOG.error("Can not find the scheduler plugin.")
             else:
diff --git a/src/core/testcase/testcase_manager.py b/src/core/testcase/testcase_manager.py
index 25c1b537fa1d5711685b4c0d53ae10676cbf1fc5..06a74865b1a777682de4ac5718bdad0db20c31a6 100644
--- a/src/core/testcase/testcase_manager.py
+++ b/src/core/testcase/testcase_manager.py
@@ -27,6 +27,8 @@ from core.common import is_open_source_product
 from core.utils import get_file_list_by_postfix
 from core.config.config_manager import FilterConfigManager
 from xdevice import platform_logger
+from xdevice import Scheduler
+from xdevice import DeviceTestType
 
 LOG = platform_logger("TestcaseManager")
 
@@ -148,6 +150,7 @@ class TestCaseManager(object):
                     suite_file_dictionary.get("PYT").append(suite_file)
                 elif suffix_name == "":
                     if file_name.find("rust") != -1:
+                        Scheduler.update_test_type_in_source("OHRust", DeviceTestType.oh_rust_test)
                         suite_file_dictionary.get("OHRust").append(suite_file)
                     else:
                         suite_file_dictionary.get("CXX").append(suite_file)