Yes, in my rushing to tidy up and get it committed, I did a last minute tweak to add this in but didn't actually commit the change back to my github fork. Looks like you've fixed it thanks!