small script fix