Fixed the setup script