diff --git a/lib/persistedretry/tagreplication/store_test.go b/lib/persistedretry/tagreplication/store_test.go
index 4ccbac94d..a95f6a11f 100644
--- a/lib/persistedretry/tagreplication/store_test.go
+++ b/lib/persistedretry/tagreplication/store_test.go
@@ -21,12 +21,12 @@ import (
 	"github.com/golang/mock/gomock"
 	"github.com/jmoiron/sqlx"
 
+	"github.com/stretchr/testify/require"
 	"github.com/uber/kraken/lib/persistedretry"
 	. "github.com/uber/kraken/lib/persistedretry/tagreplication"
 	"github.com/uber/kraken/localdb"
-	"github.com/uber/kraken/mocks/lib/persistedretry/tagreplication"
+	mocktagreplication "github.com/uber/kraken/mocks/lib/persistedretry/tagreplication"
 	"github.com/uber/kraken/utils/testutil"
-	"github.com/stretchr/testify/require"
 )
 
 type storeMocks struct {
@@ -109,7 +109,7 @@ func TestDatabaseNotLocked(t *testing.T) {
 	store := mocks.new()
 
 	var wg sync.WaitGroup
-	for i := 0; i < 1000; i++ {
+	for i := 0; i < 200; i++ {
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
diff --git a/lib/persistedretry/writeback/store_test.go b/lib/persistedretry/writeback/store_test.go
index 5377b1236..1424cd341 100644
--- a/lib/persistedretry/writeback/store_test.go
+++ b/lib/persistedretry/writeback/store_test.go
@@ -18,9 +18,9 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/require"
 	"github.com/uber/kraken/lib/persistedretry"
 	"github.com/uber/kraken/localdb"
-	"github.com/stretchr/testify/require"
 )
 
 func checkTask(t *testing.T, expected *Task, result persistedretry.Task) {
@@ -75,7 +75,7 @@ func TestDatabaseNotLocked(t *testing.T) {
 	store := NewStore(db)
 
 	var wg sync.WaitGroup
-	for i := 0; i < 1000; i++ {
+	for i := 0; i < 200; i++ {
 		wg.Add(1)
 		go func() {
 			defer wg.Done()