From 6fddcf6f8994cf38d9f3fea6aa237b642b45c46a Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <ste.pigozzi@gmail.com>
Date: Tue, 3 May 2016 21:49:17 +0200
Subject: [PATCH] ops

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 6b36c9a..122c722 100644
--- a/main.py
+++ b/main.py
@@ -188,7 +188,7 @@ class Game:
                 selected.role = Investigatore()
                 investigatore -= 1
         # Seleziona angelo
-        while investigatore > 0:
+        while angelo > 0:
             try:
                 selected = playersleft.pop()
             except IndexError:
@@ -255,7 +255,7 @@ class Game:
         for player in self.players:
             if player.alive and isinstance(player.role, Mifioso):
                 mifiosi += 1
-            elif player.alive and (isinstance(player.role, Royal) or isinstance(player.role, Investigatore)):
+            elif player.alive and player.role.team == 'Good':
                 royal += 1
         if mifiosi >= royal:
             self.message(bot, "I Mifiosi rimasti sono più dei Royal.\n"