diff --git a/main.py b/main.py
index f5c1a77..35a7a86 100644
--- a/main.py
+++ b/main.py
@@ -139,7 +139,7 @@ def zaznacz_cel_na_mapie(cel: Stan):
         krata_magazynu.krata[wiersz][kolumna] = ZawartoscPola.CEL
 
 
-def nadaj_cel_agentowi(agent: Agent, cel: Cel):
+def nadaj_cel_agentowi(agent: Agent, cel: Stan):
     agent.cel = cel
     zaznacz_cel_na_mapie(agent.cel)
     print("CEL:", agent.cel.poleStartoweGorne.wiersz, agent.cel.poleStartoweGorne.kolumna)