Other bug fix for setup script