bug fix to add