diff --git a/disttask/importinto/BUILD.bazel b/disttask/importinto/BUILD.bazel
index f2f30086debc3..5656944faea5b 100644
--- a/disttask/importinto/BUILD.bazel
+++ b/disttask/importinto/BUILD.bazel
@@ -59,18 +59,22 @@ go_test(
     timeout = "short",
     srcs = [
         "dispatcher_test.go",
+        "planner_test.go",
         "subtask_executor_test.go",
     ],
     embed = [":importinto"],
     flaky = True,
     race = "on",
+    shard_count = 4,
     deps = [
         "//br/pkg/lightning/verification",
+        "//disttask/framework/planner",
         "//disttask/framework/proto",
         "//disttask/framework/storage",
         "//domain/infosync",
         "//executor/importer",
         "//parser/model",
+        "//parser/mysql",
         "//testkit",
         "//util/logutil",
         "@com_github_ngaut_pools//:pools",