diff --git a/main.py b/main.py
index dd1b805..7f21f14 100644
--- a/main.py
+++ b/main.py
@@ -223,15 +223,13 @@ def main():
 
             algorytm = best_route(lista_paczek)
             for j in range(5):
-                Agent.cel = lista_paczek[algorytm[j]]
-                nadaj_cel_agentowi(krata_magazynu.agent, cel)
+                wiersz_cel = lista_paczek[algorytm[j]][0]
+                kolumna_cel = lista_paczek[algorytm[j]][1]
+                cel_po_algorytmu = Stan(Kierunek(random.randint(0, 3)), PoleKraty(krata_magazynu, wiersz_cel, kolumna_cel))
+                nadaj_cel_agentowi(krata_magazynu.agent, cel_po_algorytmu)
                 krata_magazynu.agent.idzDoCelu()
 
-            # powrot
-            cel = wroc()
-            nadaj_cel_agentowi(krata_magazynu.agent, cel)
-            krata_magazynu.agent.idzDoCelu()
-
+            
         if flaga1 == 1:
             osoba.krata.krata[osoba.wiersz][osoba.kolumna] = ZawartoscPola.PUSTE
             okno1.wyswietlOkno()