Fix #8
Fix #9