update for bug

update for bug