Fix chooseClass()