I took a look. I'll patch it in locally and run the test. If it passes, I'll push to master