diff --git a/wled00/data/index.js b/wled00/data/index.js
index f9bc6ae33..8136239b5 100644
--- a/wled00/data/index.js
+++ b/wled00/data/index.js
@@ -837,8 +837,6 @@ function populateSegments(s)
 	}
 
 	gId('segcont').innerHTML = cn;
-	if (simplifiedUI) gId('segcont').classList.add("simple");
-	else              gId('segcont').classList.remove("simple");
 	let noNewSegs = (lowestUnused >= maxSeg);
 	resetUtil(noNewSegs);
 	if (gId('selall')) gId('selall').checked = true;
@@ -3081,6 +3079,9 @@ function simplifyUI() {
 	// We only want Effect Search to stay on top
 	gId("palw").firstElementChild.classList.remove("staytop");
 
+	// Simplify segments
+	gId('segcont').classList.add("simple");
+
 	// Hide filter options
 	gId("filters").style.display = "none";
 }