Minor fixes for update module