diff --git a/src/autodiff/yota.jl b/src/autodiff/yota.jl
index 099145c..7b2fea5 100644
--- a/src/autodiff/yota.jl
+++ b/src/autodiff/yota.jl
@@ -1,12 +1,6 @@
 using ProximalCore
 using .Yota: grad
 
-struct YotaFunction{F}
-    f::F
-end
-
-(f::YotaFunction)(x) = f.f(x)
-
 function ProximalCore.gradient!(grad_x, f::YotaFunction, x)
     f_x, g = grad(f.f, x)
     grad_x .= g[2]
diff --git a/src/autodiff/zygote.jl b/src/autodiff/zygote.jl
index 9a724b9..c4c3262 100644
--- a/src/autodiff/zygote.jl
+++ b/src/autodiff/zygote.jl
@@ -1,12 +1,6 @@
 using ProximalCore
 using .Zygote: pullback
 
-struct ZygoteFunction{F}
-    f::F
-end
-
-(f::ZygoteFunction)(x) = f.f(x)
-
 function ProximalCore.gradient!(grad_x, f::ZygoteFunction, x)
     f_x, pb = pullback(f.f, x)
     grad_x .= pb(one(f_x))[1]