bug fix

#9