diff --git a/src/cli_interface.cpp b/src/cli_interface.cpp index bfc9f5e..0752238 100644 --- a/src/cli_interface.cpp +++ b/src/cli_interface.cpp @@ -93,7 +93,8 @@ namespace Hanabi { unsigned num_trash = 0; std::optional trash_discard_prob = 0; for(const auto &[card_multiplicity, probability]: next_states) { - if (game->is_trash(card_multiplicity.card)) { + // If the card is played, we can treat it as a trash draw as well + if (game->is_trash(card_multiplicity.card) or (play and game->cur_hand()[index] == card_multiplicity.card)) { num_trash += card_multiplicity.multiplicity; trash_discard_prob = probability; } else {