sbt graalVMVersionCheck does not fail during reload

build.sbt

@@ -47,20 +47,15 @@ val currentEdition = sys.env.getOrElse(
 val stdLibVersion       = defaultDevEnsoVersion
 val targetStdlibVersion = ensoVersion
 
-lazy val graalVMVersionCheck = taskKey[Unit]("Check GraalVM and Java versions")
-graalVMVersionCheck := {
+// Inspired by https://www.scala-sbt.org/1.x/docs/Howto-Startup.html#How+to+take+an+action+on+startup
+lazy val startupStateTransition: State => State = { s: State =>
   GraalVM.versionCheck(
     graalVersion,
     graalMavenPackagesVersion,
     javaVersion,
-    state.value.log
+    s
   )
 }
-
-// Inspired by https://www.scala-sbt.org/1.x/docs/Howto-Startup.html#How+to+take+an+action+on+startup
-lazy val startupStateTransition: State => State = { s: State =>
-  "graalVMVersionCheck" :: s
-}
 Global / onLoad := {
   val old = (Global / onLoad).value
   startupStateTransition compose old

project/GraalVM.scala

@@ -120,14 +120,15 @@ object GraalVM {
     graalVersion: String,
     graalPackagesVersion: String,
     javaVersion: String,
-    log: ManagedLogger
-  ): Unit = {
+    oldState: State
+  ): State = {
+    val log = oldState.log
     if (graalPackagesVersion != version) {
       log.error(
         s"Expected GraalVM packages version $version, but got $graalPackagesVersion. " +
         s"Version specified in build.sbt and GraalVM.scala must be in sync"
       )
-      throw new IllegalStateException("GraalVM version check failed")
+      return oldState.fail
     }
     val javaVendor = System.getProperty("java.vendor")
     if (!allowedJavaVendors.contains(javaVendor)) {
@@ -143,7 +144,7 @@ object GraalVM {
         s"Running on Java version $javaSpecVersion. " +
         s"Expected Java version $javaVersion."
       )
-      throw new IllegalStateException("java.specification.vendor check failed")
+      return oldState.fail
     }
 
     val vmVersion = System.getProperty("java.vm.version")
@@ -154,13 +155,15 @@ object GraalVM {
             s"Running on GraalVM version $version. " +
             s"Expected GraalVM version $graalVersion."
           )
-          throw new IllegalStateException("java.vm.version check failed")
+          oldState.fail
+        } else {
+          oldState
         }
       case None =>
         log.error(
           s"Could not parse GraalVM version from java.vm.version: $vmVersion."
         )
-        throw new IllegalStateException("java.vm.version check failed")
+        oldState.fail
     }
   }