Build AI fixes