small script change