small enhancement to script