Fix in setup script