diff --git a/internal/route/route_test.go b/internal/route/route_test.go
index a010fe8..3de49fe 100644
--- a/internal/route/route_test.go
+++ b/internal/route/route_test.go
@@ -33,7 +33,7 @@ func (mr *MockRouter) HandleInput(sourceId string, payload any) []route.RouteIOE
 	return nil
 }
 
-func (mr *MockRouter) HandleOutput(sourceId string, destinationId string, payload any) error {
+func (mr *MockRouter) HandleOutput(ctx context.Context, destinationId string, payload any) error {
 	return nil
 }