minor fix to script