Fix bug with setup script