diff --git a/test/Core/bayes.jl b/test/Core/bayes.jl
index b2ae78c0..ab8926f2 100644
--- a/test/Core/bayes.jl
+++ b/test/Core/bayes.jl
@@ -104,8 +104,8 @@ using Enzyme
         @test LogDensityProblems.dimension(tpostc) == length(c0)
 
         @test LogDensityProblems.capabilities(typeof(post)) === LogDensityProblems.LogDensityOrder{0}()
-        @test LogDensityProblems.capabilities(typeof(tpostf)) === LogDensityProblems.LogDensityOrder{0}()
-        @test LogDensityProblems.capabilities(typeof(tpostc)) === LogDensityProblems.LogDensityOrder{0}()
+        @test LogDensityProblems.capabilities(typeof(tpostf)) === LogDensityProblems.LogDensityOrder{1}()
+        @test LogDensityProblems.capabilities(typeof(tpostc)) === LogDensityProblems.LogDensityOrder{1}()
     end
 
     @testset "corr image prior" begin
@@ -135,7 +135,7 @@ using FiniteDifferences
     tpost = asflat(post)
 
     x = prior_sample(tpost)
-    gz = Enzyme.gradient(set_runtime_activity(Enzyme.Reverse), Const(tpost), x)
+    gz, = Enzyme.gradient(set_runtime_activity(Enzyme.Reverse), Const(tpost), x)
     mfd = central_fdm(5,1)
     gfd, = FiniteDifferences.grad(mfd, tpost, x)
     @test gz ≈ gfd
@@ -152,7 +152,7 @@ using FiniteDifferences
     x = prior_sample(tpost)
     fj = instrumentmodel(post, prior_sample(post))
     residual(post, Comrade.transform(tpost, x))
-    gz = Enzyme.gradient(set_runtime_activity(Enzyme.Reverse), Const(tpost), x)
+    gz, = Enzyme.gradient(set_runtime_activity(Enzyme.Reverse), Const(tpost), x)
     mfd = central_fdm(5,1)
     gfd, = FiniteDifferences.grad(mfd, tpost, x)
     @test gz ≈ gfd
@@ -218,7 +218,7 @@ end
         x0 = prior_sample(tpostf)
 
         @inferred logdensityof(tpostf, x0)
-        gz = Enzyme.gradient(set_runtime_activity(Enzyme.Reverse), Const(tpostf), x0)
+        gz, = Enzyme.gradient(set_runtime_activity(Enzyme.Reverse), Const(tpostf), x0)
         gn, = FiniteDifferences.grad(mfd, tpostf, x0)
         @test gz ≈ gn
     end
diff --git a/test/Core/core.jl b/test/Core/core.jl
index 1ef7e53e..7523c531 100644
--- a/test/Core/core.jl
+++ b/test/Core/core.jl
@@ -13,4 +13,4 @@ include(joinpath(@__DIR__, "observation.jl"))
 include(joinpath(@__DIR__, "partially_fixed.jl"))
 include(joinpath(@__DIR__, "models.jl"))
 include(joinpath(@__DIR__, "bayes.jl"))
-include(joinpath(@__DIR__, "rules.jl"))
+# include(joinpath(@__DIR__, "rules.jl"))
diff --git a/test/Core/partially_fixed.jl b/test/Core/partially_fixed.jl
index f8213667..d73e33e9 100644
--- a/test/Core/partially_fixed.jl
+++ b/test/Core/partially_fixed.jl
@@ -28,8 +28,8 @@ using Enzyme
         gfdf, = grad(fdm, f, x)
         gfdlj, = grad(fdm, flj, x)
 
-        gzf  = Enzyme.gradient(Enzyme.Reverse, Const(f), x)
-        gzflj = Enzyme.gradient(Enzyme.Reverse, Const(flj), x)
+        gzf,  = Enzyme.gradient(Enzyme.Reverse, Const(f), x)
+        gzflj, = Enzyme.gradient(Enzyme.Reverse, Const(flj), x)
 
         @test gzf ≈ gfdf
         @test gzflj ≈ gfdlj