minor change on script