diff --git a/src/main/kotlin/org/neoflock/neocomputers/datagen/NeoComputersDataGenerator.kt b/src/main/kotlin/org/neoflock/neocomputers/datagen/NeoComputersDataGenerator.kt
index 4a6d4a0..02db4b9 100644
--- a/src/main/kotlin/org/neoflock/neocomputers/datagen/NeoComputersDataGenerator.kt
+++ b/src/main/kotlin/org/neoflock/neocomputers/datagen/NeoComputersDataGenerator.kt
@@ -8,6 +8,6 @@ import org.neoflock.neocomputers.NeoComputers
 class NeoComputersDataGenerator : DataGeneratorEntrypoint {
     override fun onInitializeDataGenerator(generator: FabricDataGenerator) {
         val pack: FabricDataGenerator.Pack = generator.createPack()
-        pack.addProvider { o -> ModelGenerator(o as FabricDataOutput) }
+        pack.addProvider(::ModelGenerator)
     }
 }
\ No newline at end of file