fix issue #8

fix issue #8