fixed issue#8

Fixed issue#8