From 807606dae270036bf2e0eb5bfaf64edf2a49c378 Mon Sep 17 00:00:00 2001
From: Oran Nadler <orann>
Date: Tue, 17 Apr 2018 15:00:22 +0300
Subject: [PATCH] i think this is also a bug

---
 monkey_island/cc/resources/pthmap.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/monkey_island/cc/resources/pthmap.py b/monkey_island/cc/resources/pthmap.py
index 9bdfa275d..e1d544d3c 100644
--- a/monkey_island/cc/resources/pthmap.py
+++ b/monkey_island/cc/resources/pthmap.py
@@ -776,13 +776,16 @@ class PassTheHashMap(object):
 
     @cache
     def GetVictimsByAttacker(self, attacker):
+        if type(victim) != unicode:
+            victim = victim.monkey_guid
+
         victims = set()
     
         for atck, vic, _ in self.edges:
             if atck == attacker:
                 victims.add(vic)
         
-        return victims
+        return set(map(Machine, victims))
     
     @cache
     def GetInPathCountByVictim(self, victim, already_processed=None):