Made the change in recent commit. Please let me know if this is what you are looking for.