Tweak to deployment script