Incorrect solution to a problem

In PS3 (Array Logic and Data-driven Conditionals), Q8b, the solution given seems incorrect to me. The solution only works if the matrix has at least as many rows as columns (otherwise, the tally of the vector will overwrite the row tally of the matrix because of the maximum). I was looking for a solution that only uses everything learned so far but couldn't find an appropriate one.