Minor script edit