diff --git a/modules/gui.py b/modules/gui.py
index b8034bb..566972a 100644
--- a/modules/gui.py
+++ b/modules/gui.py
@@ -4,10 +4,11 @@ import pygame as pg
 from modules.agente import Agente
 from modules.mapa import Mapa
 import modules.estrategia as ModEstrategia
+
 flags = 0
 
 
-def devolveCor(agente: Agente) -> tuple[int, int, int, int]:
+def devolveCor(agente: Agente | None) -> tuple[int, int, int, int]:
     match agente.estrategia:
         case "Neutro":
             return (255, 255, 255, 0)
@@ -17,6 +18,8 @@ def devolveCor(agente: Agente) -> tuple[int, int, int, int]:
             return (255, 0, 0, 0)
         case "Lunatico":
             return (0, 0, 255, 0)
+        case None:
+            return (0, 0, 0, 0)
         case _:
             return (0, 0, 0, 0)
 
@@ -24,7 +27,7 @@ def devolveCor(agente: Agente) -> tuple[int, int, int, int]:
 def criarTabuleiro(
     surface: pg.Surface,
     dimensao_tabuleiro: tuple[int, int],
-    mapa: Mapa,
+    mapa: Mapa | None,
 ) -> None:
     dimensoes_surface_principal: tuple[int, int] = surface.get_size()
     max_altura: int = int(
@@ -119,4 +122,4 @@ def main(mapa: Mapa | None):
 
 if __name__ == '__main__':
     mapa_mundo: Mapa = Mapa((10, 10))
-    app = App(mapa_mundo)
+    app = main(mapa_mundo)