1 files changed, 29 insertions, 2 deletions
diff --git a/rules.js b/rules.js
index 6f5a5d5..1936faf 100644
--- a/rules.js
+++ b/rules.js
@@ -1048,13 +1048,27 @@ states.censorship_phase = {
 			gen_action("card", c)
 	},
 	card(c) {
+		push_undo()
 		log_br()
 		log(game.active + " - Censorship")
 		logi("C" + c)
 		discard_card(c)
+		game.state = "censorship_done"
+	},
+}
+
+states.censorship_done = {
+	inactive: "discard a card",
+	prompt() {
+		view.prompt = "Censorship Phase: Done."
+		view.actions.done = 1
+	},
+	done() {
 		game.active = enemy_player()
 		if (game.active === game.initiative)
 			goto_strategy_phase()
+		else
+			game.state = "censorship_phase"
 	},
 }
 
@@ -2003,7 +2017,7 @@ states.de_escalate_2 = {
 		let s = game.pieces[p]
 		log("Removed " + piece_abbr(p) + " from S" + s + ".")
 		remove_piece(p)
-		resume_pivotal_space_bonus_actions()
+		game.state = "pivotal_done"
 	},
 	skip() {
 		push_undo()
@@ -2038,7 +2052,7 @@ states.spread_influence = {
 		move_piece(game.who, s)
 		game.who = -1
 		if (--game.count === 0)
-			resume_pivotal_space_bonus_actions()
+			game.state = "pivotal_done"
 	},
 	done() {
 		resume_pivotal_space_bonus_actions()
@@ -2059,6 +2073,19 @@ states.turncoat = {
 		push_undo()
 		log("Replaced " + piece_abbr(p) + " in S" + s + ".")
 		replace_cube(p)
+		game.state = "pivotal_done"
+	},
+}
+
+states.pivotal_done = {
+	inactive: "confirm action",
+	prompt() {
+		let dimension = DIMENSION_SPACES[game.where]
+		view.prompt = "Pivotal Bonus Action in " + DIMENSION_NAME[game.where] + ": Done."
+		view.actions.done = 1
+	},
+	done() {
+		push_undo()
 		resume_pivotal_space_bonus_actions()
 	},
 }