Sorry been away. I craeted the checkout as a view variable in my checkout page:
return [
'product' => request()->query->get('product'),
'checkout' => request()->user()->checkout(request()->query->get('product'))
->returnTo(route('filament.management.pages.management'))
];