diff --git a/enigma-swing/src/main/java/org/quiltmc/enigma/gui/Gui.java b/enigma-swing/src/main/java/org/quiltmc/enigma/gui/Gui.java
index 58791c22..8cd3e755 100644
--- a/enigma-swing/src/main/java/org/quiltmc/enigma/gui/Gui.java
+++ b/enigma-swing/src/main/java/org/quiltmc/enigma/gui/Gui.java
@@ -55,6 +55,7 @@
 import javax.swing.JScrollBar;
 import javax.swing.JSplitPane;
 import javax.swing.SwingUtilities;
+import javax.swing.UIManager;
 import javax.swing.WindowConstants;
 import java.awt.BorderLayout;
 import java.awt.Container;
@@ -162,6 +163,9 @@ private void setupDockers() {
 	}
 
 	private void setupUi() {
+		// fix folder icons being automatically hidden: https://github.com/JFormDesigner/FlatLaf/pull/609
+		UIManager.put("Tree.showDefaultIcons", true);
+
 		this.setupDockers();
 
 		this.jarFileChooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
diff --git a/enigma-swing/src/main/java/org/quiltmc/enigma/gui/element/ClassTreeCellRenderer.java b/enigma-swing/src/main/java/org/quiltmc/enigma/gui/element/ClassTreeCellRenderer.java
index aa514498..3e2148ff 100644
--- a/enigma-swing/src/main/java/org/quiltmc/enigma/gui/element/ClassTreeCellRenderer.java
+++ b/enigma-swing/src/main/java/org/quiltmc/enigma/gui/element/ClassTreeCellRenderer.java
@@ -15,7 +15,6 @@
 import javax.swing.Icon;
 import javax.swing.JLabel;
 import javax.swing.JTree;
-import javax.swing.UIManager;
 import javax.swing.tree.DefaultTreeCellRenderer;
 import java.awt.Component;
 import java.util.function.Function;
@@ -27,9 +26,6 @@ public class ClassTreeCellRenderer extends DefaultTreeCellRenderer {
 	public ClassTreeCellRenderer(Gui gui, ClassSelector selector) {
 		this.controller = gui.getController();
 		this.selector = selector;
-
-		// fix folder icons being automatically hidden: https://github.com/JFormDesigner/FlatLaf/pull/609
-		UIManager.put("Tree.showDefaultIcons", true);
 	}
 
 	@Override