Ahh, crap I forgot to update the logic here for the general case. I will fix that.