diff --git a/pkg/lightning/backend/backend_test.go b/pkg/lightning/backend/backend_test.go
index de462353f..64297af24 100644
--- a/pkg/lightning/backend/backend_test.go
+++ b/pkg/lightning/backend/backend_test.go
@@ -11,7 +11,7 @@ import (
 	"github.com/pingcap/parser/mysql"
 
 	kv "github.com/pingcap/br/pkg/lightning/backend"
-	"github.com/pingcap/br/pkg/mock"
+	"github.com/pingcap/br/pkg/lightning/mock"
 )
 
 type backendSuite struct {
diff --git a/pkg/lightning/backend/importer_test.go b/pkg/lightning/backend/importer_test.go
index 36f95ba96..5819e2da9 100644
--- a/pkg/lightning/backend/importer_test.go
+++ b/pkg/lightning/backend/importer_test.go
@@ -28,7 +28,7 @@ import (
 
 	kv "github.com/pingcap/br/pkg/lightning/backend"
 	"github.com/pingcap/br/pkg/lightning/common"
-	"github.com/pingcap/br/pkg/mock"
+	"github.com/pingcap/br/pkg/lightning/mock"
 )
 
 type importerSuite struct {
diff --git a/pkg/mock/backend.go b/pkg/lightning/mock/backend.go
similarity index 100%
rename from pkg/mock/backend.go
rename to pkg/lightning/mock/backend.go
diff --git a/pkg/mock/glue.go b/pkg/lightning/mock/glue.go
similarity index 100%
rename from pkg/mock/glue.go
rename to pkg/lightning/mock/glue.go
diff --git a/pkg/mock/glue_checkpoint.go b/pkg/lightning/mock/glue_checkpoint.go
similarity index 100%
rename from pkg/mock/glue_checkpoint.go
rename to pkg/lightning/mock/glue_checkpoint.go
diff --git a/pkg/mock/importer.go b/pkg/lightning/mock/importer.go
similarity index 100%
rename from pkg/mock/importer.go
rename to pkg/lightning/mock/importer.go
diff --git a/pkg/mock/storage.go b/pkg/lightning/mock/storage.go
similarity index 100%
rename from pkg/mock/storage.go
rename to pkg/lightning/mock/storage.go
diff --git a/pkg/lightning/mydump/reader_test.go b/pkg/lightning/mydump/reader_test.go
index 90d3f3967..7785c86b1 100644
--- a/pkg/lightning/mydump/reader_test.go
+++ b/pkg/lightning/mydump/reader_test.go
@@ -22,8 +22,8 @@ import (
 	"github.com/golang/mock/gomock"
 	. "github.com/pingcap/check"
 
+	"github.com/pingcap/br/pkg/lightning/mock"
 	. "github.com/pingcap/br/pkg/lightning/mydump"
-	"github.com/pingcap/br/pkg/mock"
 	"github.com/pingcap/br/pkg/storage"
 )
 
diff --git a/pkg/lightning/restore/restore_test.go b/pkg/lightning/restore/restore_test.go
index bf38e691f..c32760827 100644
--- a/pkg/lightning/restore/restore_test.go
+++ b/pkg/lightning/restore/restore_test.go
@@ -42,10 +42,10 @@ import (
 	"github.com/pingcap/br/pkg/lightning/config"
 	"github.com/pingcap/br/pkg/lightning/glue"
 	"github.com/pingcap/br/pkg/lightning/log"
+	"github.com/pingcap/br/pkg/lightning/mock"
 	"github.com/pingcap/br/pkg/lightning/mydump"
 	"github.com/pingcap/br/pkg/lightning/verification"
 	"github.com/pingcap/br/pkg/lightning/worker"
-	"github.com/pingcap/br/pkg/mock"
 	"github.com/pingcap/br/pkg/storage"
 )