diff --git a/src/YM/YMflow.jl b/src/YM/YMflow.jl
index 3287e29..5103d90 100644
--- a/src/YM/YMflow.jl
+++ b/src/YM/YMflow.jl
@@ -407,7 +407,7 @@ function krnl_add_qd!(rm, op, frc1, frc2, U, lp::SpaceParm{4,M,B,D}) where {M,B,
     return nothing
 end
 
-function krnl_field_tensor!(frc1, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, ztw1, ztw2, lp::SpaceParm{4,M,B,D}) where {T,M,B,D}
+function krnl_field_tensor!(frc1::AbstractArray{TA}, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, ztw1, ztw2, lp::SpaceParm{4,M,B,D}) where {TA,T,M,B,D}
 
     b, r = CUDA.threadIdx().x, CUDA.blockIdx().x
     it = point_time((b,r), lp)
@@ -446,8 +446,8 @@ function krnl_field_tensor!(frc1, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, z
 
     if SFBC && (it == lp.iL[end])
         frc1[b,1,r]     = projalg(Ush[b,1]*l1/Ush[b,2])
-        frc1[bu1,2,ru1] = zero(frc1[bu1,2,ru1])
-        frc1[bd,3,rd]   = zero(frc1[bu1,2,ru1])
+        frc1[bu1,2,ru1] = zero(TA)
+        frc1[bd,3,rd]   = zero(TA)
         frc1[bu2,4,ru2] = projalg(l2*l1)
     else
         frc1[b,1,r]     = projalg(ztw1, Ush[b,1]*l1/Ush[b,2])
@@ -487,8 +487,8 @@ function krnl_field_tensor!(frc1, frc2, U::AbstractArray{T}, Ubnd, ipl1, ipl2, z
 
     if SFBC && (it == lp.iL[end])
         frc2[b,1,r]     = projalg(Ush[b,1]*l1/Ush[b,2])
-        frc1[bu1,2,ru1] = zero(frc1[bu1,2,ru1])
-        frc1[bd,3,rd]   = zero(frc1[bu1,2,ru1])
+        frc1[bu1,2,ru1] = zero(TA)
+        frc1[bd,3,rd]   = zero(TA)
         frc2[bu2,4,ru2] = projalg(l2*l1)
     else
         frc2[b,1,r]     = projalg(ztw1, Ush[b,1]*l1/Ush[b,2])