diff --git a/destination/producer_mock.go b/destination/producer_mock.go
index 5027aee..ed8f86e 100644
--- a/destination/producer_mock.go
+++ b/destination/producer_mock.go
@@ -21,6 +21,7 @@ import (
 type MockProducer struct {
 	ctrl     *gomock.Controller
 	recorder *MockProducerMockRecorder
+	isgomock struct{}
 }
 
 // MockProducerMockRecorder is the mock recorder for MockProducer.
diff --git a/go.mod b/go.mod
index 365a8fe..aa771c8 100644
--- a/go.mod
+++ b/go.mod
@@ -14,7 +14,7 @@ require (
 	github.com/rs/zerolog v1.33.0
 	github.com/twmb/franz-go v1.18.0
 	github.com/twmb/franz-go/pkg/kadm v1.14.0
-	go.uber.org/mock v0.4.0
+	go.uber.org/mock v0.5.0
 )
 
 require (
diff --git a/go.sum b/go.sum
index e8be733..d127123 100644
--- a/go.sum
+++ b/go.sum
@@ -488,8 +488,8 @@ go.uber.org/automaxprocs v1.5.3 h1:kWazyxZUrS3Gs4qUpbwo5kEIMGe/DAvi5Z4tl2NW4j8=
 go.uber.org/automaxprocs v1.5.3/go.mod h1:eRbA25aqJrxAbsLO0xy5jVwPt7FQnRgjW+efnwa1WM0=
 go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
 go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
-go.uber.org/mock v0.4.0 h1:VcM4ZOtdbR4f6VXfiOpwpVJDL6lCReaZ6mw31wqh7KU=
-go.uber.org/mock v0.4.0/go.mod h1:a6FSlNadKUHUa9IP5Vyt1zh4fC7uAwxMutEAscFbkZc=
+go.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=
+go.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=
 go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
 go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
 go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8=
diff --git a/source/consumer_mock.go b/source/consumer_mock.go
index cc24176..8a0ccd0 100644
--- a/source/consumer_mock.go
+++ b/source/consumer_mock.go
@@ -20,6 +20,7 @@ import (
 type MockConsumer struct {
 	ctrl     *gomock.Controller
 	recorder *MockConsumerMockRecorder
+	isgomock struct{}
 }
 
 // MockConsumerMockRecorder is the mock recorder for MockConsumer.
diff --git a/source/franz_mock.go b/source/franz_mock.go
index b756d7c..45c2ec4 100644
--- a/source/franz_mock.go
+++ b/source/franz_mock.go
@@ -21,6 +21,7 @@ import (
 type MockClient struct {
 	ctrl     *gomock.Controller
 	recorder *MockClientMockRecorder
+	isgomock struct{}
 }
 
 // MockClientMockRecorder is the mock recorder for MockClient.
@@ -77,10 +78,10 @@ func (c *MockClientCloseCall) DoAndReturn(f func()) *MockClientCloseCall {
 }
 
 // CommitRecords mocks base method.
-func (m *MockClient) CommitRecords(arg0 context.Context, arg1 ...*kgo.Record) error {
+func (m *MockClient) CommitRecords(ctx context.Context, rs ...*kgo.Record) error {
 	m.ctrl.T.Helper()
-	varargs := []any{arg0}
-	for _, a := range arg1 {
+	varargs := []any{ctx}
+	for _, a := range rs {
 		varargs = append(varargs, a)
 	}
 	ret := m.ctrl.Call(m, "CommitRecords", varargs...)
@@ -89,9 +90,9 @@ func (m *MockClient) CommitRecords(arg0 context.Context, arg1 ...*kgo.Record) er
 }
 
 // CommitRecords indicates an expected call of CommitRecords.
-func (mr *MockClientMockRecorder) CommitRecords(arg0 any, arg1 ...any) *MockClientCommitRecordsCall {
+func (mr *MockClientMockRecorder) CommitRecords(ctx any, rs ...any) *MockClientCommitRecordsCall {
 	mr.mock.ctrl.T.Helper()
-	varargs := append([]any{arg0}, arg1...)
+	varargs := append([]any{ctx}, rs...)
 	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CommitRecords", reflect.TypeOf((*MockClient)(nil).CommitRecords), varargs...)
 	return &MockClientCommitRecordsCall{Call: call}
 }
@@ -120,17 +121,17 @@ func (c *MockClientCommitRecordsCall) DoAndReturn(f func(context.Context, ...*kg
 }
 
 // OptValue mocks base method.
-func (m *MockClient) OptValue(arg0 any) any {
+func (m *MockClient) OptValue(opt any) any {
 	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "OptValue", arg0)
+	ret := m.ctrl.Call(m, "OptValue", opt)
 	ret0, _ := ret[0].(any)
 	return ret0
 }
 
 // OptValue indicates an expected call of OptValue.
-func (mr *MockClientMockRecorder) OptValue(arg0 any) *MockClientOptValueCall {
+func (mr *MockClientMockRecorder) OptValue(opt any) *MockClientOptValueCall {
 	mr.mock.ctrl.T.Helper()
-	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OptValue", reflect.TypeOf((*MockClient)(nil).OptValue), arg0)
+	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OptValue", reflect.TypeOf((*MockClient)(nil).OptValue), opt)
 	return &MockClientOptValueCall{Call: call}
 }
 
@@ -158,17 +159,17 @@ func (c *MockClientOptValueCall) DoAndReturn(f func(any) any) *MockClientOptValu
 }
 
 // PollFetches mocks base method.
-func (m *MockClient) PollFetches(arg0 context.Context) kgo.Fetches {
+func (m *MockClient) PollFetches(ctx context.Context) kgo.Fetches {
 	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "PollFetches", arg0)
+	ret := m.ctrl.Call(m, "PollFetches", ctx)
 	ret0, _ := ret[0].(kgo.Fetches)
 	return ret0
 }
 
 // PollFetches indicates an expected call of PollFetches.
-func (mr *MockClientMockRecorder) PollFetches(arg0 any) *MockClientPollFetchesCall {
+func (mr *MockClientMockRecorder) PollFetches(ctx any) *MockClientPollFetchesCall {
 	mr.mock.ctrl.T.Helper()
-	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PollFetches", reflect.TypeOf((*MockClient)(nil).PollFetches), arg0)
+	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PollFetches", reflect.TypeOf((*MockClient)(nil).PollFetches), ctx)
 	return &MockClientPollFetchesCall{Call: call}
 }