1 files changed, 2 insertions, 4 deletions

diff --git a/src/main/java/cuchaz/enigma/config/Themes.java b/src/main/java/cuchaz/enigma/config/Themes.java
index b3132f2..8a4c341 100644
--- a/src/main/java/cuchaz/enigma/config/Themes.java
+++ b/src/main/java/cuchaz/enigma/config/Themes.java
@@ -33,9 +33,7 @@ public class Themes {
         gui.obfuscatedHighlightPainter = new ObfuscatedHighlightPainter();
         gui.deobfuscatedHighlightPainter = new DeobfuscatedHighlightPainter();
         gui.otherHighlightPainter = new OtherHighlightPainter();
-        gui.editor.updateUI();
-        gui.editor.setBackground(new Color(Config.getInstance().editorBackground));
-            SwingUtilities.updateComponentTreeUI(gui.getFrame());
-        gui.getController().refreshCurrentClass();
+        gui.setEditorTheme(Config.getInstance().lookAndFeel);
+        SwingUtilities.updateComponentTreeUI(gui.getFrame());
     }
 }