Change contains and assert to in and @assert