Conditional Eager loading issue

    public static function table(Table $table): Table
    {
        return $table
            ->modifyQueryUsing(function ($query){
                $query->with([
                    'memberable',
                    'memberCreatedBy',
                    'memberUpdatedBy',
                    'platformTeamMemberAssignedBy',
                    'assignedPlatformTeamMembers',
                ]);

                    // Eager load specific relationships based on memberable_type
                $query->when($query->getQuery()->where('memberable_type', Company::class)->exists(), function ($query) {
                    $query->with(['memberable.profile', 'memberable.profile.addresses']);
                });

                $query->when($query->getQuery()->where('memberable_type', User::class)->exists(), function ($query) {
                    $query->with('memberable.addresses');
                });


                return $query;

            })
            ->columns([