Minor script change