A small script tweak