Fix AI to follow new protocol