fix issue #9

fix issue #9