Minor script tweak