This seemed a little messy so let me know if there's a better way to test for this.