Fix the robot code