Fix #9

Fix #9