add bug fix