Yep. I'll fix and add a test for that.