Fix update stuff