diff --git a/code/tests/_game_test_puppeteer.dm b/code/tests/_game_test_puppeteer.dm
index 130a6938158..008dd733ab7 100644
--- a/code/tests/_game_test_puppeteer.dm
+++ b/code/tests/_game_test_puppeteer.dm
@@ -184,3 +184,9 @@
 
 /datum/test_puppeteer/proc/drop_held_item()
 	puppet.drop_item_to_ground(puppet.get_active_hand())
+
+/datum/test_puppeteer/proc/add_trait(trait)
+	ADD_TRAIT(puppet, trait, TRAIT_GENERIC)
+
+/datum/test_puppeteer/proc/remove_trait(trait)
+	REMOVE_TRAIT(puppet, trait, TRAIT_GENERIC)
diff --git a/code/tests/attack_chain/test_attack_chain_structures.dm b/code/tests/attack_chain/test_attack_chain_structures.dm
index 3042e522ea2..e47d759c53b 100644
--- a/code/tests/attack_chain/test_attack_chain_structures.dm
+++ b/code/tests/attack_chain/test_attack_chain_structures.dm
@@ -19,6 +19,7 @@
 /datum/game_test/room_test/attack_chain_structures/Run()
 	var/datum/test_puppeteer/player = new(src)
 	player.puppet.name = "Player"
+	player.add_trait(TRAIT_FLASH_PROTECTION)
 
 	player.puppet.mind.add_antag_datum(/datum/antagonist/cultist)
 	player.spawn_obj_in_hand(/obj/item/melee/cultblade/dagger)