Setup script fix