Minor correction to script