Fix Ceausescu and TST Award interaction

1 files changed, 9 insertions, 5 deletions

diff --git a/rules.js b/rules.js
index 5a880bc..fa86dc2 100644
--- a/rules.js
+++ b/rules.js
@@ -4563,10 +4563,11 @@ function vm_return() {
 	delete game.vm_influence_added
 	delete game.communist_hand_red
 	game.vm_event = 0
+
 	if (game.persistent_events.includes(C_AUSTRIA_HUNGARY_BORDER_REOPENED)) {
 		reset_austria_hungary_border_reopened()
 	}
-	if (game.is_pwr_struggle || game.state === 'vm_tst_6' || game.return_state === 'ceausescu_check') {
+	if (game.is_pwr_struggle || game.state === 'vm_tst_6' || game.state === 'vm_tst_3' || game.return_state === 'ceausescu_check') {
 		vm_end_event()
 	} else if ((is_auto_resolve(game.played_card) && game.played_card !== C_KISS_OF_DEATH ) &&
 			((cards[game.played_card].side === 'C' && game.active === DEM) ||
@@ -8106,6 +8107,9 @@ states.vm_tst_3 = {
 			for (let card of game.valid_cards) {
 				gen_action_card(card)
 			}
+		} else {
+			view.prompt = 'Discard cards: Done.'
+			gen_action('done')
 		}
 	},
 	card(card) {
@@ -8113,11 +8117,11 @@ states.vm_tst_3 = {
 		discard(card)
 		game.valid_cards.splice(game.valid_cards.indexOf(card), 1)
 		game.temp ++
-		if (game.temp === 2) {
-			game.valid_cards = []
-			vm_next()
-		}
 	},
+	done() {
+		game.valid_cards = []
+		vm_next()
+	}
 }
 
 states.vm_tst_4 = {