Minor fixes to update process