little script change