diff --git a/pkg/gateway/serverless_handlers_test.go b/pkg/gateway/serverless_handlers_test.go
index aacf655..9d97d78 100644
--- a/pkg/gateway/serverless_handlers_test.go
+++ b/pkg/gateway/serverless_handlers_test.go
@@ -16,8 +16,8 @@ type mockFunctionRegistry struct {
 	functions []*serverless.Function
 }
 
-func (m *mockFunctionRegistry) Register(ctx context.Context, fn *serverless.FunctionDefinition, wasmBytes []byte) error {
-	return nil
+func (m *mockFunctionRegistry) Register(ctx context.Context, fn *serverless.FunctionDefinition, wasmBytes []byte) (*serverless.Function, error) {
+	return nil, nil
 }
 
 func (m *mockFunctionRegistry) Get(ctx context.Context, namespace, name string, version int) (*serverless.Function, error) {
@@ -36,6 +36,10 @@ func (m *mockFunctionRegistry) GetWASMBytes(ctx context.Context, wasmCID string)
 	return []byte("wasm"), nil
 }
 
+func (m *mockFunctionRegistry) GetLogs(ctx context.Context, namespace, name string, limit int) ([]serverless.LogEntry, error) {
+	return []serverless.LogEntry{}, nil
+}
+
 func TestServerlessHandlers_ListFunctions(t *testing.T) {
 	logger := zap.NewNop()
 	registry := &mockFunctionRegistry{