diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 2659871..2fa9097 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -53,8 +53,10 @@ jobs:
       - name: Compile examples module
         run: sbt "project examples" compile
 
-      - name: Check if docs are up to date
+      - name: Check if docsRoot are up to date
         run: sbt "docsRoot/mdoc --check"
+
+      - name: Check if docs are up to date
         run: sbt "docs/mdoc --check"
         
       - name: Run tests
diff --git a/build.sbt b/build.sbt
index e4af855..e3d09c1 100644
--- a/build.sbt
+++ b/build.sbt
@@ -91,6 +91,7 @@ lazy val docsRoot = (project in file(".dimwit-docs-root"))
       "VERSION" -> version.value
     ),
     fork := true,
+    javaOptions ++= scalapyJavaOptions,
     envVars := (ThisBuild / envVars).value
   )
 
@@ -107,5 +108,6 @@ lazy val docs = (project in file(".dimwit-docs"))
       "VERSION" -> version.value
     ),
     fork := true,
+    javaOptions ++= scalapyJavaOptions,
     envVars := (ThisBuild / envVars).value
   )
diff --git a/core/src/test/scala/dimwit/package.scala b/core/src/test/scala/dimwit/package.scala
index 0d797a7..192772e 100644
--- a/core/src/test/scala/dimwit/package.scala
+++ b/core/src/test/scala/dimwit/package.scala
@@ -42,7 +42,7 @@ object MustBeFloat:
   given MustBeFloat[Float] with {}
 
   transparent inline given [V]: MustBeFloat[V] =
-    error("approxEqual can only be used with Float tensors. For Int tensors, use 'equal(...)'.") 
+    error("approxEqual can only be used with Float tensors. For Int tensors, use 'equal(...)'.")
 
 private lazy val _dimwitTestInit: Unit = dimwit.initialize()
 
diff --git a/core/src/test/scala/dimwit/tensor/TensorOpsStructureSuite.scala b/core/src/test/scala/dimwit/tensor/TensorOpsStructureSuite.scala
index 52a0ee1..daaee22 100644
--- a/core/src/test/scala/dimwit/tensor/TensorOpsStructureSuite.scala
+++ b/core/src/test/scala/dimwit/tensor/TensorOpsStructureSuite.scala
@@ -81,7 +81,7 @@ class TensorOpsStructureSuite extends DimwitTest:
         val acbDimCode = "t.rearrange((Axis[A |*| C], Axis[B |*| D]), Axis[A] -> 2, Axis[C] -> 2, Axis[B] -> 2)"
         val acbDimErrors = typeCheckErrors(acbDimCode)
         acbDimErrors should have size 1
-        acbDimErrors.head.message should not include ("Missing Axis: 'dimwit.D'")
+        acbDimErrors.head.message should include("Missing Axis: 'dimwit.D'")
 
         "t.rearrange((Axis[A |*| C], Axis[B |*| D]), Axis[A] -> 2, Axis[C] -> 2, Axis[B] -> 2, Axis[D] -> 2)" should compile