Add 2-doc comparison assert and fix caller