update - fixed bug