diff --git a/Project.toml b/Project.toml
index 7508ed3..3c4bb83 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "Checkpointing"
 uuid = "eb46d486-4f9c-4c3d-b445-a617f2a2f1ca"
 authors = ["Michel Schanen <mschanen@anl.gov>", "Sri Hari Krishna Narayanan <snarayan@mcs.anl.gov>"]
-version = "0.9.5"
+version = "0.9.6"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
@@ -14,7 +14,7 @@ Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
 [compat]
 ChainRulesCore = "1"
 DataStructures = "0.18"
-Enzyme = "0.11, 0.12"
+Enzyme = "0.11, 0.12, 0.13"
 HDF5 = "0.16, 0.17"
 julia = "1.8"
 
diff --git a/src/Rules/EnzymeRules.jl b/src/Rules/EnzymeRules.jl
index 28f360a..0bb7caf 100644
--- a/src/Rules/EnzymeRules.jl
+++ b/src/Rules/EnzymeRules.jl
@@ -21,7 +21,7 @@ function augmented_primal(
 end
 
 function reverse(
-    config::ConfigWidth{1},
+    config,
     ::Const{typeof(Checkpointing.checkpoint_struct_for)},
     dret::Type{<:Const},
     tape,
@@ -60,7 +60,7 @@ function augmented_primal(
 end
 
 function reverse(
-    config::ConfigWidth{1},
+    config,
     ::Const{typeof(Checkpointing.checkpoint_struct_while)},
     dret::Type{<:Const},
     tape,