diff --git a/pkg/checkout/mutation_cancel_payment.go b/pkg/checkout/mutation_cancel_payment.go
index 44a6b22..7be9d62 100644
--- a/pkg/checkout/mutation_cancel_payment.go
+++ b/pkg/checkout/mutation_cancel_payment.go
@@ -7,7 +7,7 @@ import (
 	messages "git.k6n.net/go-cart-actor/proto/checkout"
 )
 
-func HandlePaymentCancelled(g *CheckoutGrain, m *messages.PaymentDeclined) error {
+func HandlePaymentCancelled(g *CheckoutGrain, m *messages.CancelPayment) error {
 
 	payment, found := g.FindPayment(m.PaymentId)
 	if !found {