Fix update scripts