Fixing up setup script.