Another script tweak