update for fixed bug