diff --git a/pkg/ddl/tests/tiflash/ddl_tiflash_test.go b/pkg/ddl/tests/tiflash/ddl_tiflash_test.go
index 16f5ca8b32647..3f0a1e2822a1d 100644
--- a/pkg/ddl/tests/tiflash/ddl_tiflash_test.go
+++ b/pkg/ddl/tests/tiflash/ddl_tiflash_test.go
@@ -1400,7 +1400,7 @@ func TestTiFlashAvailableAfterAddPartition(t *testing.T) {
 func TestTiFlashAvailableAfterDownOneStore(t *testing.T) {
 	s, teardown := createTiFlashContext(t)
 	defer teardown()
-	tk := testkit.NewTestKit(t, s.store)
+	tk := testkit.NewTestKitWithSession(t, s.store, testkit.NewSession(t, s.store))
 
 	tk.MustExec("use test")
 	tk.MustExec("drop table if exists ddltiflash")