1 files changed, 1 insertions, 3 deletions
diff --git a/ui.js b/ui.js
index 7b2c336..5e93d4a 100644
--- a/ui.js
+++ b/ui.js
@@ -565,9 +565,7 @@ function update_map() {
 			let steps = game.known[b][1];
 			let moved = game.known[b][2];
 			let element = ui.known[b];
-			let color = BLOCKS[b].owner;
-			if (color == REBEL)
-				color = game.pretender;
+			let color = block_owner(b);
 
 			show_block(element);
 			layout[area][color].push(element);