update for bug  fix