tiny bug fix