diff --git a/.github/workflows/Documentation.yml b/.github/workflows/Documentation.yml
index 37c9b69d..5f8def4e 100644
--- a/.github/workflows/Documentation.yml
+++ b/.github/workflows/Documentation.yml
@@ -24,9 +24,9 @@ jobs:
           sudo apt-get install libhdf5-dev
           sudo apt-get install --reinstall libxcb-xinerama0
           sudo apt-get install qtbase5-dev
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v3
       - name: Setup python
-        uses: actions/setup-python@v1
+        uses: actions/setup-python@v2
         with:
           python-version: ${{ matrix.python-version }}
           architecture: ${{ matrix.arch }}
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index c9e0d8c4..f28b039e 100755
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -39,9 +39,9 @@ jobs:
           sudo apt-get update
           sudo apt-get install libhdf5-dev
           sudo apt-get install --reinstall libxcb-xinerama0
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v3
       - name: Setup python
-        uses: actions/setup-python@v1
+        uses: actions/setup-python@v2
         with:
           python-version: ${{ matrix.python-version }}
           architecture: ${{ matrix.arch }}
@@ -51,7 +51,7 @@ jobs:
           version: ${{ matrix.version }}
           arch: ${{ matrix.arch }}
       - uses: julia-actions/julia-buildpkg@v1      
-      - uses: actions/cache@v1
+      - uses: actions/cache@v3
         env:
           cache-name: cache-artifacts
         with:
diff --git a/lib/ComradeOptimization/Project.toml b/lib/ComradeOptimization/Project.toml
index 98fa6ca6..f47ffa41 100644
--- a/lib/ComradeOptimization/Project.toml
+++ b/lib/ComradeOptimization/Project.toml
@@ -28,6 +28,7 @@ OptimizationOptimJL = "36348300-93cb-4f02-beb5-3c3902f8871e"
 Pyehtim = "3d61700d-6e5b-419a-8e22-9c066cf00468"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 VLBIImagePriors = "b1ba175b-8447-452c-b961-7db2d6f7a029"
+Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [targets]
-test = ["Test", "OptimizationBBO", "OptimizationOptimJL", "Pyehtim", "Distributions", "VLBIImagePriors"]
+test = ["Test", "OptimizationBBO", "OptimizationOptimJL", "Pyehtim", "Distributions", "VLBIImagePriors", "Zygote"]
diff --git a/lib/ComradeOptimization/test/runtests.jl b/lib/ComradeOptimization/test/runtests.jl
index 088c2f83..ef821f9f 100644
--- a/lib/ComradeOptimization/test/runtests.jl
+++ b/lib/ComradeOptimization/test/runtests.jl
@@ -1,4 +1,5 @@
 using Pyehtim, Comrade, ComradeOptimization, OptimizationOptimJL, Distributions, VLBIImagePriors
+using Zygote
 using Test
 
 include(joinpath(@__DIR__, "../../../test/test_util.jl"))
diff --git a/test/Core/bayes.jl b/test/Core/bayes.jl
index 210ec952..e7a97663 100644
--- a/test/Core/bayes.jl
+++ b/test/Core/bayes.jl
@@ -6,6 +6,7 @@ using Plots
 using LogDensityProblems
 using LogDensityProblemsAD
 using Pyehtim
+using Zygote
 
 
 
@@ -33,7 +34,7 @@ using Pyehtim
     @test dimension(post) == dimension(tpostf)
     @test dimension(post) == dimension(tpostc)
 
-    f = OptimizationFunction(tpostc, Optimization.Zygote())
+    f = OptimizationFunction(tpostc, Optimization.AutoZygote())
     x0 = [ 0.1,
            0.4,
            0.5,