diff --git a/Assemblies/AIImages.dll b/Assemblies/AIImages.dll
index a0812a7..0db4f3d 100644
Binary files a/Assemblies/AIImages.dll and b/Assemblies/AIImages.dll differ
diff --git a/Source/AIImages/PawnGizmoPatch.cs b/Source/AIImages/PawnGizmoPatch.cs
index 8c81a97..f5854fb 100644
--- a/Source/AIImages/PawnGizmoPatch.cs
+++ b/Source/AIImages/PawnGizmoPatch.cs
@@ -10,7 +10,7 @@ using Verse;
 namespace AIImages
 {
     /// <summary>
-    /// Harmony patch to add a gizmo (button) to all colonist pawns
+    /// Harmony patch to add a gizmo (button) to all pawns
     /// </summary>
     [HarmonyPatch(typeof(Pawn), "GetGizmos")]
     [System.Diagnostics.CodeAnalysis.SuppressMessage(
@@ -34,12 +34,8 @@ namespace AIImages
                 yield return gizmo;
             }
 
-            // Only add button to colonist pawns that are spawned
-            if (
-                __instance.IsColonist
-                && __instance.Spawned
-                && __instance.Faction == Faction.OfPlayer
-            )
+            // Add button to all spawned pawns (not just colonists)
+            if (__instance.Spawned)
             {
                 yield return new Command_Action
                 {
diff --git a/Source/AIImages/Window_AIImage.cs b/Source/AIImages/Window_AIImage.cs
index c1807d0..384aaed 100644
--- a/Source/AIImages/Window_AIImage.cs
+++ b/Source/AIImages/Window_AIImage.cs
@@ -272,11 +272,9 @@ namespace AIImages
             base.WindowUpdate();
 
             // Проверяем, изменилась ли выбранная пешка
-            Pawn selectedPawn = Find.Selector.SelectedPawns.FirstOrDefault(p =>
-                p.IsColonist && p.Spawned && p.Faction == Faction.OfPlayer
-            );
+            Pawn selectedPawn = Find.Selector.SelectedPawns.FirstOrDefault(p => p.Spawned);
 
-            // Если выбрана новая колонистская пешка, обновляем окно
+            // Если выбрана новая пешка, обновляем окно
             if (selectedPawn != null && selectedPawn != pawn)
             {
                 UpdatePawn(selectedPawn);