Fix domain specialization integration bug