I thought this was a reasonable comment @shoyer - a little disappointed this didn't make it in in the new PR at #980. I suppose I can also see the argument for keeping the change simple, but I also think this would have been a more robust check.