Update to script so it doesn't fail on first run