diff --git a/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt b/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt
index 065d1b7..4d54139 100644
--- a/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt
+++ b/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt
@@ -21,11 +21,11 @@ class ForgeInit(container: ModContainer): Snowy() {
         .allScanData
         .flatMap { obj: ModFileScanData -> obj.classes }
         .filter { data: ModFileScanData.ClassData? ->
-            (data!!.javaClass.getDeclaredField("clazz").apply { isAccessible = true }.get(data) as Type).className.let { classname ->
-                classname.startsWith(FeaturePackage) && Class.forName(classname).isValidForButtonCollection()
+            data!!.clazz.className.let {
+                it.startsWith(FeaturePackage) && Class.forName(it).isValidForButtonCollection()
             }
         }
-        .map { Class.forName((it!!.javaClass.getDeclaredField("clazz").apply { isAccessible = true }.get(it) as Type).className).asSubclass(ButtonImpl::class.java)}
+        .map { Class.forName((it!!.clazz.className)).asSubclass(ButtonImpl::class.java)}
         .toSet()
     init {
         container.eventBus!!.run {