Setup script tweak