diff --git a/tests/e2e/offline_inference/test_moss_tts_nano.py b/tests/e2e/offline_inference/test_moss_tts_nano.py
index 5596654ef..40561d53f 100644
--- a/tests/e2e/offline_inference/test_moss_tts_nano.py
+++ b/tests/e2e/offline_inference/test_moss_tts_nano.py
@@ -8,7 +8,7 @@ import pytest
 import torch
 from vllm import SamplingParams
 
-from tests.utils import hardware_test
+from tests.helpers.mark import hardware_test
 from vllm_omni import Omni
 
 MODEL_NAME = "OpenMOSS-Team/MOSS-TTS-Nano"
diff --git a/tests/e2e/online_serving/test_moss_tts_nano.py b/tests/e2e/online_serving/test_moss_tts_nano.py
index ec30bd4da..22e305345 100644
--- a/tests/e2e/online_serving/test_moss_tts_nano.py
+++ b/tests/e2e/online_serving/test_moss_tts_nano.py
@@ -14,9 +14,9 @@ os.environ["VLLM_TEST_CLEAN_GPU_MEMORY"] = "0"
 
 import pytest
 
-from tests.conftest import OmniServerParams
+from tests.helpers.mark import hardware_test
+from tests.helpers.runtime import OmniServerParams
 from tests.helpers.stage_config import get_deploy_config_path
-from tests.utils import hardware_test
 
 MODEL = "OpenMOSS-Team/MOSS-TTS-Nano"