Fix mock payment flow