minor change to script