diff --git a/mocks/mockKVClient.go b/mocks/mockKVClient.go
index 456d196..2343994 100644
--- a/mocks/mockKVClient.go
+++ b/mocks/mockKVClient.go
@@ -222,7 +222,7 @@
 }
 
 // Watch mock function implementation for KVClient
-func (kvclient *MockKVClient) Watch(ctx context.Context, key string) chan *kvstore.Event {
+func (kvclient *MockKVClient) Watch(ctx context.Context, key string, withPrefix bool) chan *kvstore.Event {
 	return nil
 	// if key == "" {
 	// 	return nil