diff --git a/addon.txt b/addon.txt
index 8a825d8..3b534e7 100644
--- a/addon.txt
+++ b/addon.txt
@@ -16,8 +16,12 @@ If you want your enemies to be more human, this mod will change all hostile NPC
 - Change the behavior of custom models NPCs from combine to citizen
 - Allows hostile to attack with melee weapons
 
----
-
 I'll make VJ version if i can
 
+---
+
+models used:
+- Lenna https://steamcommunity.com/sharedfiles/filedetails/?id=3167885628
+- Tomoko https://steamcommunity.com/sharedfiles/filedetails/?id=3133369749
+
 Note: i couldn't find a mod similar to the one i made, so i don't know who made it first.
\ No newline at end of file
diff --git a/lua/autorun/hostile_npcs_tocitizen.lua b/lua/autorun/hostile_npcs_tocitizen.lua
index 48e1880..9415020 100644
--- a/lua/autorun/hostile_npcs_tocitizen.lua
+++ b/lua/autorun/hostile_npcs_tocitizen.lua
@@ -1,17 +1,16 @@
-local npcs = list.Get("NPC")
-local citizenEnemy = npcs["npc_citizen_rebel_enemy"]
+local originalListSet = list.Set
 
-if citizenEnemy then
-    for class, npc in pairs(npcs) do
+function list.Set(listType, class, npc)
+    if listType == "NPC" then
         if npc["Class"] == "npc_combine_s" and npc["Category"] ~= "#spawnmenu.category.combine" then
+            local citizenEnemy = list.Get("NPC")["npc_citizen_rebel_enemy"]
             npc["Class"] = citizenEnemy["Class"]
             npc["KeyValues"] = {
                 Hostile = 1,
                 SquadName =	"overwatch",
                 citizentype = 4
             }
-        
-            list.Set("NPC", class, npc)
         end
     end
-end
+    originalListSet(listType, class, npc)
+end
\ No newline at end of file