Fixes bug in update script.