bug fix
#8 fixed