Fix bug in setup script