Update changes + bug fix.