Possible bug fix for diag_mult() operations