bug fix

tiny bugs