Minor script fix