Fix to update script for before install