Update handling of TST

1 files changed, 4 insertions, 2 deletions

diff --git a/rules.js b/rules.js
index dd25ceb..7abbfab 100644
--- a/rules.js
+++ b/rules.js
@@ -2889,7 +2889,8 @@ function do_tst_attempt() {
 
 	// TIANANMEN SQUARE ATTEMPT
 	game.return = game.active
-	game.return_state = 'tiananmen_square_attempt_done'
+
+	//game.return_state = 'tiananmen_square_attempt_done'
 	if (game.active === DEM) {
 		game.dem_tst_attempted_this_turn = 1
 		if (roll >= dem_tst_req[game.dem_tst_position]) {
@@ -4761,7 +4762,7 @@ function vm_return() {
 	}
 
 	// Check if end event state is needed
-	if (game.is_pwr_struggle || game.state === 'vm_tst_6' || game.return_state === 'tiananmen_square_attempt_done' || game.return_state === 'ceausescu_check') {
+	if (game.is_pwr_struggle || game.state === 'vm_tst_6' || game.return_state === 'ceausescu_check') {
 		vm_end_event()
 	}
 	// Check if auto-resolve opponent event
@@ -7286,6 +7287,7 @@ states.vm_goodbye_lenin_ops = {
 	},
 	tst() {
 		push_undo()
+		game.return = game.round_player
 		game.state = 'vm_tiananmen_square_attempt'
 	},
 }