Fix robot logic