1 files changed, 5 insertions, 3 deletions
diff --git a/src/main/kotlin/com/dulkirfabric/config/DulkirConfig.kt b/src/main/kotlin/com/dulkirfabric/config/DulkirConfig.kt
index 5e2eb5d..93db1a0 100644
--- a/src/main/kotlin/com/dulkirfabric/config/DulkirConfig.kt
+++ b/src/main/kotlin/com/dulkirfabric/config/DulkirConfig.kt
@@ -38,9 +38,9 @@ class DulkirConfig {
 
     private val buttonText: Text =
         MutableText.of(LiteralTextContent("Dulkir")).formatted(Formatting.BOLD, Formatting.YELLOW)
-    var screen: Screen
+    var screen: Screen = buildScreen(null)
 
-    init {
+    fun buildScreen(parentScreen: Screen? = null): Screen {
         val builder = ConfigBuilder.create().setTitle(buttonText)
         builder.setDefaultBackgroundTexture(Identifier("minecraft:textures/block/oak_planks.png"))
         builder.setGlobalized(true)
@@ -408,7 +408,9 @@ class DulkirConfig {
 //                .build()
 //        )
         builder.transparentBackground()
-        screen = builder.build()
+        if (parentScreen != null)
+            builder.setParentScreen(parentScreen)
+        return builder.build()
     }
 
     @Serializable