small script tweak