diff --git a/Project.toml b/Project.toml
index 82241d9..37f5ea6 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "Boltz"
 uuid = "4544d5e4-abc5-4dea-817f-29e4c205d9c8"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "0.3.6"
+version = "0.3.7"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
diff --git a/src/layers/hamiltonian.jl b/src/layers/hamiltonian.jl
index 2656c71..944d9ba 100644
--- a/src/layers/hamiltonian.jl
+++ b/src/layers/hamiltonian.jl
@@ -10,7 +10,7 @@ position and momentum.
 ## Arguments
 
   - `FST`: If `true`, then the type of the state returned by the model must be same as the
-    type of the input state. See the documentation on [`StatefulLayer`](@ref) for more
+    type of the input state. See the documentation on [`StatefulLuxLayer`](@ref) for more
     information.
   - `model`: A `Lux.AbstractExplicitLayer` neural network that returns the Hamiltonian of
     the system. The `model` must return a "batched scalar", i.e. all the dimensions of the