Fix update/install bugs