fix issus #3

fix issus #3