diff --git a/src/script/civitaiHelperFix.ts b/src/script/civitaiHelperFix.ts index c7a7d05..252daeb 100644 --- a/src/script/civitaiHelperFix.ts +++ b/src/script/civitaiHelperFix.ts @@ -91,10 +91,20 @@ const updateCardForCivitai = () => { metadataButton = card.querySelector('.metadata-button'); // Additional node additionalNode = card.querySelector('.actions .additional'); + // Get ul node, which is the parent of all buttons ulNode = card.querySelector('.actions .additional ul'); + if (ulNode == null) { + ulNode = document.createElement("ul"); + additionalNode.appendChild(ulNode); + } + // Replace preview text button replacePreviewButton = card.querySelector('.actions .additional a'); + if (replacePreviewButton == null) { + replacePreviewButton = document.createElement("a"); + additionalNode.appendChild(replacePreviewButton); + } // Check thumb mode if (isThumbMode && additionalNode) {