update for bug fix