diff --git a/docs/src/.use.md.swp b/docs/src/.use.md.swp
deleted file mode 100644
index 2de26b5..0000000
Binary files a/docs/src/.use.md.swp and /dev/null differ
diff --git a/src/Operators.jl b/src/Operators.jl
index 1a8fb70..3ae56ca 100644
--- a/src/Operators.jl
+++ b/src/Operators.jl
@@ -57,7 +57,7 @@ struct Shrink{T} <: Operator
     plan::T
 
     function Shrink(threshold, sigma, state)
-        s = size(state.recipSpace)
+        s = size(state.realSpace)
         kernel = zeros(ComplexF64, s)
         for i in 1:s[1]
             for j in 1:s[2]
diff --git a/src/State.jl b/src/State.jl
index 74b755b..7788621 100644
--- a/src/State.jl
+++ b/src/State.jl
@@ -31,6 +31,7 @@ struct State{T}
         shift .= [1,1,1] .- round.(Int64, mapreduce(sqrt, +, intens))
         shift .*= -1
         intens = CuArray{Float64, 3, CUDA.Mem.DeviceBuffer}(circshift(intens,shift))
+        recSupport = CuArray{Float64, 3, CUDA.Mem.DeviceBuffer}(circshift(recSupport,shift))
 
         realSpace = CUDA.zeros(ComplexF64, s)
         state = BcdiCore.TradState("L2", false, realSpace, intens, recSupport)