Fix matrix multiplication part for DNS.