Cool, I'll add that in. I kind of wish there was a more thorough (and easy to implement) way to test this.