diff --git a/lib/ComradeOptimization/test/runtests.jl b/lib/ComradeOptimization/test/runtests.jl
index 57a96ca4..cd6817f2 100644
--- a/lib/ComradeOptimization/test/runtests.jl
+++ b/lib/ComradeOptimization/test/runtests.jl
@@ -10,7 +10,7 @@ include(joinpath(@__DIR__, "../../../test/test_util.jl"))
     post = Posterior(lklhd, prior)
 
     tpost = ascube(post)
-    f = OptimizationFunction(tpost, Optimization.AutoForwardDiff{4}())
+    f = OptimizationFunction(tpost, Optimization.AutoForwardDiff())
     x0 = [ 0.1,
            0.4,
            0.5,
diff --git a/test/Core/bayes.jl b/test/Core/bayes.jl
index a2ba2816..7d5b95e9 100644
--- a/test/Core/bayes.jl
+++ b/test/Core/bayes.jl
@@ -33,7 +33,7 @@ using Pyehtim
     @test dimension(post) == dimension(tpostf)
     @test dimension(post) == dimension(tpostc)
 
-    f = OptimizationFunction(tpostc, Optimization.AutoForwardDiff{4}())
+    f = OptimizationFunction(tpostc, Optimization.AutoForwardDiff())
     x0 = [ 0.1,
            0.4,
            0.5,