In reply to Ciprian.

Thank you Ciprian. I've accepted the pull request and updated it on WordPress.org.

Should've probably had that initially, but looking over each other's code is what open source is all about. :)