diff --git a/core/src/main/scala/dimwit/tensor/Tensor.scala b/core/src/main/scala/dimwit/tensor/Tensor.scala
index 027dba50..6ea683f9 100644
--- a/core/src/main/scala/dimwit/tensor/Tensor.scala
+++ b/core/src/main/scala/dimwit/tensor/Tensor.scala
@@ -72,7 +72,7 @@ class Tensor[T <: Tuple: Labels, V] private[dimwit] (
   def extent[L: Label](axis: Axis[L])(using ev: AxisIndex[T, L]): AxisExtent[L] =
     shape.extent(axis)
 
-  private val jaxTypeName: String = py.Dynamic.global.`type`(jaxValue).`__name__`.as[String]
+  private lazy val jaxTypeName: String = py.Dynamic.global.`type`(jaxValue).`__name__`.as[String]
   lazy val isTracer: Boolean =
     val jaxCoreTracer = py.module("jax.core").Tracer
     py.Dynamic.global.isinstance(jaxValue, jaxCoreTracer).as[Boolean]