Fixing bug - Some HTML not updated