clearly with shape check as you have it. That way you avoid all fancy indexing.