AI registration bug fixed