Update to fix #23 Thanks @richlove1