Updated bug fix