diff --git a/python/sglang/multimodal_gen/csrc/attn/vmoba_attn/tests/test_vmoba_attn.py b/python/sglang/multimodal_gen/csrc/attn/vmoba_attn/tests/test_vmoba_attn.py
index f4304bda4..9350f3c9e 100644
--- a/python/sglang/multimodal_gen/csrc/attn/vmoba_attn/tests/test_vmoba_attn.py
+++ b/python/sglang/multimodal_gen/csrc/attn/vmoba_attn/tests/test_vmoba_attn.py
@@ -4,8 +4,7 @@ import random
 
 import pytest
 import torch
-from csrc.attn.vmoba_attn.vmoba import moba_attn_varlen
-
+from sglang.multimodal_gen.csrc.attn.vmoba_attn.vmoba import moba_attn_varlen
 
 def generate_test_data(
     batch_size, total_seqlen, num_heads, head_dim, dtype, device="cuda"
diff --git a/test/manual/nightly/test_deepseek_v31_perf.py b/test/manual/nightly/test_deepseek_v31_perf.py
index 68d57b646..b04cfb6d2 100644
--- a/test/manual/nightly/test_deepseek_v31_perf.py
+++ b/test/manual/nightly/test_deepseek_v31_perf.py
@@ -1,7 +1,6 @@
 import unittest
 
-from nightly_utils import NightlyBenchmarkRunner
-
+from sglang.test.nightly_utils import NightlyBenchmarkRunner
 from sglang.test.test_utils import DEFAULT_URL_FOR_TEST, _parse_int_list_env
 
 DEEPSEEK_V31_MODEL_PATH = "deepseek-ai/DeepSeek-V3.1"
diff --git a/test/manual/nightly/test_deepseek_v32_perf.py b/test/manual/nightly/test_deepseek_v32_perf.py
index 20ba0c125..b14a189ed 100644
--- a/test/manual/nightly/test_deepseek_v32_perf.py
+++ b/test/manual/nightly/test_deepseek_v32_perf.py
@@ -1,7 +1,6 @@
 import unittest
 
-from nightly_utils import NightlyBenchmarkRunner
-
+from sglang.test.nightly_utils import NightlyBenchmarkRunner
 from sglang.test.test_utils import DEFAULT_URL_FOR_TEST, _parse_int_list_env
 
 DEEPSEEK_V32_MODEL_PATH = "deepseek-ai/DeepSeek-V3.2-Exp"
diff --git a/test/manual/nightly/test_text_models_perf.py b/test/manual/nightly/test_text_models_perf.py
index 1e2cf70ff..f83483ac2 100644
--- a/test/manual/nightly/test_text_models_perf.py
+++ b/test/manual/nightly/test_text_models_perf.py
@@ -1,7 +1,6 @@
 import unittest
 
-from nightly_utils import NightlyBenchmarkRunner
-
+from sglang.test.nightly_utils import NightlyBenchmarkRunner
 from sglang.test.test_utils import (
     DEFAULT_URL_FOR_TEST,
     ModelLaunchSettings,
diff --git a/test/manual/nightly/test_vlms_perf.py b/test/manual/nightly/test_vlms_perf.py
index b837c0262..0d4c2801b 100644
--- a/test/manual/nightly/test_vlms_perf.py
+++ b/test/manual/nightly/test_vlms_perf.py
@@ -2,8 +2,7 @@ import os
 import unittest
 import warnings
 
-from nightly_utils import NightlyBenchmarkRunner
-
+from sglang.test.nightly_utils import NightlyBenchmarkRunner
 from sglang.test.test_utils import (
     DEFAULT_URL_FOR_TEST,
     ModelLaunchSettings,
diff --git a/test/manual/test_two_batch_overlap.py b/test/manual/test_two_batch_overlap.py
index 7c8bc7eb1..410872166 100644
--- a/test/manual/test_two_batch_overlap.py
+++ b/test/manual/test_two_batch_overlap.py
@@ -3,12 +3,12 @@ from types import SimpleNamespace
 
 import requests
 
-from sglang.srt.environ import envs
-from sglang.srt.model_executor.forward_batch_info import ForwardMode
-from sglang.srt.two_batch_overlap import (
+from sglang.srt.batch_overlap.two_batch_overlap import (
     compute_split_seq_index,
     compute_split_token_index,
 )
+from sglang.srt.environ import envs
+from sglang.srt.model_executor.forward_batch_info import ForwardMode
 from sglang.srt.utils import kill_process_tree
 from sglang.test.run_eval import run_eval
 from sglang.test.test_utils import (