Minor change to script