Fix #11

Thanks for submitting issue
Fix #12