diff --git a/src/gradle-tooling-extension/groovy/com/demonwav/mcdev/platform/mcp/gradle/tooling/neomoddev/NeoModDevGradleModelBuilderImpl.groovy b/src/gradle-tooling-extension/groovy/com/demonwav/mcdev/platform/mcp/gradle/tooling/neomoddev/NeoModDevGradleModelBuilderImpl.groovy
index f443c5a8f..ff8385ccc 100644
--- a/src/gradle-tooling-extension/groovy/com/demonwav/mcdev/platform/mcp/gradle/tooling/neomoddev/NeoModDevGradleModelBuilderImpl.groovy
+++ b/src/gradle-tooling-extension/groovy/com/demonwav/mcdev/platform/mcp/gradle/tooling/neomoddev/NeoModDevGradleModelBuilderImpl.groovy
@@ -51,7 +51,14 @@ final class NeoModDevGradleModelBuilderImpl implements ModelBuilderService {
 
         def neoFormVersion
         try {
-            neoFormVersion = extension.neoFormVersion.getOrNull()
+            def neoFormVersionProp = extension.neoFormVersion
+            if (neoFormVersionProp instanceof String) {
+                neoFormVersion = neoFormVersionProp
+            } else if (neoFormVersionProp instanceof Provider) {
+                neoFormVersion = neoFormVersionProp.getOrNull()
+            } else {
+                neoFormVersion = null
+            }
         } catch (InvalidUserCodeException ignore) {
             // Happens when the NeoForm version is not set
             neoFormVersion = null