diff --git a/src/utils/device.jl b/src/utils/device.jl
index 891b19f..627095b 100644
--- a/src/utils/device.jl
+++ b/src/utils/device.jl
@@ -17,7 +17,12 @@ send_to_device(
     ::Val{:gpu},
     x::Union{
         SubArray{<:Any,<:Any,<:Union{CircularArrayBuffer,ElasticArray}},
-        Base.ReshapedArray{<:Any,<:Any,<:SubArray{<:Any,<:Any,<:CircularArrayBuffer}},
+        Base.ReshapedArray{
+            <:Any,
+            <:Any,
+            <:SubArray{<:Any,<:Any,<:Union{CircularArrayBuffer,ElasticArray}},
+        },
+        Base.ReshapedArray{<:Any,<:Any,<:Union{CircularArrayBuffer,ElasticArray}},
         SubArray{
             <:Any,
             <:Any,