diff --git a/main.py b/main.py
index e97ac54..0bdd36a 100644
--- a/main.py
+++ b/main.py
@@ -11,7 +11,7 @@ from domain.entities.vacuum import Vacuum
 from domain.entities.docking_station import Doc_Station
 from domain.world import World
 from view.renderer import Renderer
-# from AI_brain.movement import StateGraphSearchBFS, State
+# from AI_brain.movement import GoAnyDirectionBFS, State
 from AI_brain.rotate_and_go_bfs import RotateAndGoBFS, State