diff --git a/src/array/setindex.jl b/src/array/setindex.jl
index a7170e1c6..1d8290c47 100644
--- a/src/array/setindex.jl
+++ b/src/array/setindex.jl
@@ -1,4 +1,3 @@
-
 export setindex
 
 struct SetIndex{T,N} <: ArrayOp{T,N}
@@ -41,6 +40,6 @@ function stage(ctx::Context, sidx::SetIndex)
             q
         end
     end
-    inp.chunks = ps
-    inp
+
+    return DArray(eltype(inp), d, subdmns, ps, inp.partitioning)
 end