Tweak to the install script