Fix regression for non-square state multiplication.