Fixed bug in setup script