tests/peers-bootstrapping/main.go | 9 ++++++---

diff --git a/tests/peers-bootstrapping/main.go b/tests/peers-bootstrapping/main.go
index 4114c5c385655a1309af1e29cd8996517d18743c..2e0b3822208bc9f7599f9a33063693333b12877e 100644
--- a/tests/peers-bootstrapping/main.go
+++ b/tests/peers-bootstrapping/main.go
@@ -9,7 +9,7 @@ 	"os"
 	"path/filepath"
 	"time"
 
-	_ "github.com/anacrolix/envpprof"
+	"github.com/anacrolix/envpprof"
 	"github.com/anacrolix/log"
 	"github.com/anacrolix/sync"
 	"github.com/dustin/go-humanize"
@@ -18,6 +18,7 @@
 	"github.com/anacrolix/torrent"
 	"github.com/anacrolix/torrent/bencode"
 	"github.com/anacrolix/torrent/metainfo"
+	"github.com/anacrolix/torrent/storage"
 )
 
 func assertNil(x any) {
@@ -37,6 +38,7 @@ 	return cfg
 }
 
 func main() {
+	defer envpprof.Stop()
 	tmpDir, err := os.MkdirTemp("", "peers-bootstrapping")
 	assertNil(err)
 	slog.Info("made temp dir", slog.String("tmpDir", tmpDir))
@@ -56,7 +58,7 @@ 	assertNil(err)
 	var clients []*torrent.Client
 	var torrents []*torrent.Torrent
 	clientConfig := newClientConfig()
-	clientConfig.DataDir = sourceDir
+	clientConfig.DefaultStorage = storage.NewMMap(sourceDir)
 	initialClient, err := torrent.NewClient(clientConfig)
 	assertNil(err)
 	clientIndex := 0
@@ -91,7 +93,7 @@ 	for range 5 {
 		clientIndex := clientIndex
 		storageDir := filepath.Join(tmpDir, fmt.Sprintf("client%v", clientIndex))
 		clientConfig := newClientConfig()
-		clientConfig.DataDir = storageDir
+		clientConfig.DefaultStorage = storage.NewMMap(storageDir)
 		clientConfig.Logger = log.Default.WithValues(slog.Int("clientIndex", clientIndex))
 		//clientConfig.Logger.Levelf(log.Critical, "test")
 		client, err := torrent.NewClient(clientConfig)
@@ -132,4 +134,5 @@ 			humanize.Bytes(uint64(written.Int64())),
 			humanize.Bytes(uint64(read.Int64())),
 		)
 	}
+	assertNil(os.RemoveAll(tmpDir))
 }