small change bug fix