diff --git a/docs/Manifest.toml b/docs/Manifest.toml
index bee958b..db4c866 100644
--- a/docs/Manifest.toml
+++ b/docs/Manifest.toml
@@ -2,7 +2,7 @@
 
 julia_version = "1.10.2"
 manifest_format = "2.0"
-project_hash = "fc42f77a62c3f6eb111e0b5b1979b582dffac6c0"
+project_hash = "9952708dc4e30427802feb8e3372b2b9577fe53d"
 
 [[deps.ADTypes]]
 git-tree-sha1 = "016833eb52ba2d6bea9fcb50ca295980e728ee24"
diff --git a/docs/make.jl b/docs/make.jl
index bc4ce95..a96d96c 100644
--- a/docs/make.jl
+++ b/docs/make.jl
@@ -1,5 +1,11 @@
 using Documenter
 using TrixiEnzyme
+import Trixi
+import Enzyme
+import Polyester
+
+DocMeta.setdocmeta!(TrixiEnzyme, :DocTestSetup, :(using TrixiEnzyme); recursive=true)
+
 
 push!(LOAD_PATH,"../src/")
 makedocs(sitename="TrixiEnzyme.jl",
diff --git a/src/TrixiEnzyme.jl b/src/TrixiEnzyme.jl
index 7f657f1..51a7dd9 100644
--- a/src/TrixiEnzyme.jl
+++ b/src/TrixiEnzyme.jl
@@ -14,6 +14,7 @@ using Trixi: AbstractEquations, TreeMesh, DGSEM,
              wrap_array, compute_coefficients, have_nonconservative_terms,
              boundary_condition_periodic,
              set_log_type, set_sqrt_type
+import Enzyme
 using Enzyme: autodiff, Forward, Reverse, Duplicated, DuplicatedNoNeed, make_zero
 using Polyester: @batch
 
diff --git a/src/jacobian.jl b/src/jacobian.jl
index 2a18083..58ee199 100644
--- a/src/jacobian.jl
+++ b/src/jacobian.jl
@@ -65,6 +65,7 @@ function jacobian_enzyme_reverse_closure(semi)
     end
 
     return dxs
+end
 
 """
     jacobian_enzyme_forward(semi::SemidiscretizationHyperbolic)
diff --git a/test/runtests.jl b/test/runtests.jl
index f9c9ccb..132aacf 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,4 +1,3 @@
 using Test, TrixiEnzyme
-out = plusTwo(3)
 
 @test out == 5