diff --git a/common/src/main/kotlin/fr/username404/snowygui/gui/feature/ButtonImpl.kt b/common/src/main/kotlin/fr/username404/snowygui/gui/feature/ButtonImpl.kt
index de33aa2..857fa6a 100644
--- a/common/src/main/kotlin/fr/username404/snowygui/gui/feature/ButtonImpl.kt
+++ b/common/src/main/kotlin/fr/username404/snowygui/gui/feature/ButtonImpl.kt
@@ -28,12 +28,12 @@ sealed class ButtonImpl: ColoredElement(0.0, 0.0, 73, 8, opacity = 0.60F) {
                                     }
                                 }
                             }
-                            box.name!!.key == impl.info.parent.translationKey
+                            box.name?.key ?: box.name.toString() == impl.info.parent.translationKey
                         }
                     }
                 }.entries.forEach { entry ->
                     with(entry) {
-                        entry.value.forEach {
+                        value.forEach {
                             key?.buttons!!.add(it)
                         }
                     }