Editing Spatie Tags

Forms\Components\TextInput::make('name')
                            ->afterStateHydrated(function (Forms\Components\TextInput $component, \Spatie\Tags\Tag $tag, Forms\Components\Livewire $livewire) {
                                $component->state($tag->getTranslation('name', $livewire->activeLocale));
                    })

Forms\Components\TextInput::make('name')
                            ->afterStateHydrated(function (Forms\Components\TextInput $component, \Spatie\Tags\Tag $tag, Forms\Components\Livewire $livewire) {
                                $component->state($tag->getTranslation('name', $livewire->activeLocale));
                    })

App\Filament\Resources\TagResource::App\Filament\Resources\{closure}(): Argument #3 ($livewire) must be of type Filament\Forms\Components\Livewire, App\Filament\Resources\TagResource\Pages\ManageTags given, called in /Users/xxx/xxx/xxx/xxx/vendor/filament/support/src/Concerns/EvaluatesClosures.php on line 35

App\Filament\Resources\TagResource::App\Filament\Resources\{closure}(): Argument #3 ($livewire) must be of type Filament\Forms\Components\Livewire, App\Filament\Resources\TagResource\Pages\ManageTags given, called in /Users/xxx/xxx/xxx/xxx/vendor/filament/support/src/Concerns/EvaluatesClosures.php on line 35