Small change to script