Minor fix in choose_pit

Fixed the return of choose pit for player 2