From 8b9be2072c2294b35cd08532e63b3493c07d6b25 Mon Sep 17 00:00:00 2001
From: "Hoang, Minh Duc" <minh-duc.hoang@fernuni-hagen.de>
Date: Fri, 24 Oct 2025 17:22:10 +0200
Subject: [PATCH] #36 Comparing the display order should only be necessary if a
 category with flavors is present

---
 amd/src/ui.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/amd/src/ui.js b/amd/src/ui.js
index e843d4b..39ddf83 100644
--- a/amd/src/ui.js
+++ b/amd/src/ui.js
@@ -165,7 +165,8 @@ const displayDialogue = async(editor) => {
     if (soleCategories.length > 0 || selectCategories.length > 0) {
         const savedCategory = currentCategoryId;
         const savedFlavor = lastFlavor[currentCategoryId];
-        if (soleCategories.length == 0 || soleCategories[0].displayorder > selectCategories[0].displayorder) {
+        if ((selectCategories.length > 0)
+            && (soleCategories.length == 0 || soleCategories[0].displayorder > selectCategories[0].displayorder)) {
             selectCategories[0].click();
         } else {
             soleCategories[0].click();
@@ -236,6 +237,11 @@ const handleCategoryFlavorClick = (event, modal) => {
 
     const componentButtons = modal.getRoot()[0].querySelectorAll('.elements-buttons-preview button');
     componentButtons.forEach(componentButton => {
+        // Remove previous flavor.
+        if (componentButton.dataset.flavor != undefined) {
+            componentButton.classList.remove(componentButton.dataset.flavor);
+        }
+        componentButton.classList.add(currentFlavor);
         componentButton.dataset.flavor = currentFlavor;
         if (
             (componentButton.dataset.flavorlist == '' || componentButton.dataset.flavorlist.split(',').includes(currentFlavor)) &&