Small fix in setup script