diff --git a/src/flash/tms470.c b/src/flash/tms470.c
index 65e7016f9..008961139 100644
--- a/src/flash/tms470.c
+++ b/src/flash/tms470.c
@@ -436,6 +436,7 @@ int tms470_unlock_flash(struct flash_bank_s *bank)
 
 	if (keysSet)
 	{
+		key_set_count = 5;
 		p_key_sets[0] = flashKeys;
 		p_key_sets[1] = FLASH_KEYS_ALL_ONES;
 		p_key_sets[2] = FLASH_KEYS_ALL_ZEROS;