diff --git a/src/primitives.jl b/src/primitives.jl
index b9ed18e7..25b1869f 100644
--- a/src/primitives.jl
+++ b/src/primitives.jl
@@ -1019,14 +1019,14 @@ end
 
 function exchange_impl!(rcv,snd,graph,setup,::Type{T}) where T<:AbstractVector
     @assert is_consistent(graph)
-    @assert eltype(rcv) <: JaggedArray
+    @assert eltype(rcv) <: AbstractJaggedArray
     snd_ids = graph.snd
     rcv_ids = graph.rcv
     @assert length(rcv_ids) == length(rcv)
     @assert length(rcv_ids) == length(snd)
     for rcv_id in 1:length(rcv_ids)
         for (i, snd_id) in enumerate(rcv_ids[rcv_id])
-            snd_snd_id = JaggedArray(snd[snd_id])
+            snd_snd_id = jagged_array(snd[snd_id])
             j = first(findall(k->k==rcv_id,snd_ids[snd_id]))
             ptrs_rcv = rcv[rcv_id].ptrs
             ptrs_snd = snd_snd_id.ptrs