One more script fix