diff --git a/pieceio/mocks/PieceIO.go b/pieceio/mocks/PieceIO.go
index baac4ddb3..0f7d9f2bd 100644
--- a/pieceio/mocks/PieceIO.go
+++ b/pieceio/mocks/PieceIO.go
@@ -2,43 +2,45 @@
 
 package mocks
 
-import cid "github.com/ipfs/go-cid"
-import filestore "github.com/filecoin-project/go-fil-components/filestore"
-import io "io"
-import ipld "github.com/ipld/go-ipld-prime"
-import mock "github.com/stretchr/testify/mock"
-import pieceio "github.com/filecoin-project/go-fil-components/pieceio"
+import (
+	io "io"
+
+	filestore "github.com/filecoin-project/go-fil-components/filestore"
+	cid "github.com/ipfs/go-cid"
+
+	ipld "github.com/ipld/go-ipld-prime"
+
+	mock "github.com/stretchr/testify/mock"
+)
 
 // PieceIO is an autogenerated mock type for the PieceIO type
 type PieceIO struct {
 	mock.Mock
 }
 
-// GeneratePieceCommitment provides a mock function with given fields: bs, payloadCid, selector
-func (_m *PieceIO) GeneratePieceCommitment(bs pieceio.ReadStore, payloadCid cid.Cid, selector ipld.Node) ([]byte, filestore.File, error) {
-	ret := _m.Called(bs, payloadCid, selector)
+// GeneratePieceCommitment provides a mock function with given fields: payloadCid, selector
+func (_m *PieceIO) GeneratePieceCommitment(payloadCid cid.Cid, selector ipld.Node) ([]byte, filestore.Path, error) {
+	ret := _m.Called(payloadCid, selector)
 
 	var r0 []byte
-	if rf, ok := ret.Get(0).(func(pieceio.ReadStore, cid.Cid, ipld.Node) []byte); ok {
-		r0 = rf(bs, payloadCid, selector)
+	if rf, ok := ret.Get(0).(func(cid.Cid, ipld.Node) []byte); ok {
+		r0 = rf(payloadCid, selector)
 	} else {
 		if ret.Get(0) != nil {
 			r0 = ret.Get(0).([]byte)
 		}
 	}
 
-	var r1 filestore.File
-	if rf, ok := ret.Get(1).(func(pieceio.ReadStore, cid.Cid, ipld.Node) filestore.File); ok {
-		r1 = rf(bs, payloadCid, selector)
+	var r1 filestore.Path
+	if rf, ok := ret.Get(1).(func(cid.Cid, ipld.Node) filestore.Path); ok {
+		r1 = rf(payloadCid, selector)
 	} else {
-		if ret.Get(1) != nil {
-			r1 = ret.Get(1).(filestore.File)
-		}
+		r1 = ret.Get(1).(filestore.Path)
 	}
 
 	var r2 error
-	if rf, ok := ret.Get(2).(func(pieceio.ReadStore, cid.Cid, ipld.Node) error); ok {
-		r2 = rf(bs, payloadCid, selector)
+	if rf, ok := ret.Get(2).(func(cid.Cid, ipld.Node) error); ok {
+		r2 = rf(payloadCid, selector)
 	} else {
 		r2 = ret.Error(2)
 	}
@@ -46,20 +48,20 @@ func (_m *PieceIO) GeneratePieceCommitment(bs pieceio.ReadStore, payloadCid cid.
 	return r0, r1, r2
 }
 
-// ReadPiece provides a mock function with given fields: r, bs
-func (_m *PieceIO) ReadPiece(r io.Reader, bs pieceio.WriteStore) (cid.Cid, error) {
-	ret := _m.Called(r, bs)
+// ReadPiece provides a mock function with given fields: r
+func (_m *PieceIO) ReadPiece(r io.Reader) (cid.Cid, error) {
+	ret := _m.Called(r)
 
 	var r0 cid.Cid
-	if rf, ok := ret.Get(0).(func(io.Reader, pieceio.WriteStore) cid.Cid); ok {
-		r0 = rf(r, bs)
+	if rf, ok := ret.Get(0).(func(io.Reader) cid.Cid); ok {
+		r0 = rf(r)
 	} else {
 		r0 = ret.Get(0).(cid.Cid)
 	}
 
 	var r1 error
-	if rf, ok := ret.Get(1).(func(io.Reader, pieceio.WriteStore) error); ok {
-		r1 = rf(r, bs)
+	if rf, ok := ret.Get(1).(func(io.Reader) error); ok {
+		r1 = rf(r)
 	} else {
 		r1 = ret.Error(1)
 	}