diff --git a/src/chambolle.jl b/src/chambolle.jl
index 4f7fa7b36670f44271d0f909c83ee10f0494e3d2..370e7c4b7eeab2b714554b24cc8f1a372fb47a01 100644
--- a/src/chambolle.jl
+++ b/src/chambolle.jl
@@ -52,7 +52,7 @@ function init(md::ChambolleModel, alg::ChambolleAlgorithm)
     sv = zero(rv)
     p = extend(reshape(reinterpret(SVector{d,Float64}, pv), size(md.g)), StaticKernels.ExtensionNothing())
     r = reshape(reinterpret(Float64, rv), size(md.g))
-    s = extend(reshape(reinterpret(Float64, sv), size(md.g)), StaticKernels.ExtensionConstant(0.))
+    s = extend(reshape(reinterpret(Float64, sv), size(md.g)), StaticKernels.ExtensionReplicate())
 
     z = zero(CartesianIndex{d})
     @inline kf1(pw, gw) = @inbounds divergence(pw) + gw[z]