minor fix on script