Updte script fix