Update bug fix