This is an interesting point... I guess `.argmax` should return the argument of the maximum in the *unmasked* data, then?  I think that's a valid interpretation, so this is just a :+1: and acknowledgement of the change.