diff --git a/src/layers.jl b/src/layers.jl
index 06ebe05..4206a2f 100644
--- a/src/layers.jl
+++ b/src/layers.jl
@@ -245,16 +245,6 @@ function ipa_customgrad(ipa::Union{IPCrossA, IPA}, TiL::Tuple{AbstractArray,Abst
     @assert size(mask,1) == size(siR, 2)
     @assert size(mask,2) == size(siL, 2)
     @assert siL == siR && TiL == TiR
-
-    if zij != nothing
-        #This is assuming the dims of zij are c, N_frames_L, N_frames_R, batch
-        @assert size(zij,2) == size(siR,2)
-        @assert size(zij,3) == size(siL,2) 
-    end
-    if mask != 0
-        @assert size(mask,1) == size(siR, 2)
-        @assert size(mask,2) == size(siL, 2)
-    end
     
     # Get relevant parameters from our ipa struct.
     l = ipa.layers