Hi Paul.
My goodness, you are absolutely correct. This is lousy assert checking on my end. Thanks for pointing it out :) i will have to edit this :)
The same logic should work for small straight. Sorted set where we check [0] and [4]
Thanks again for reaching out! This is why I love Medium :)