little script fix