bug fix after update