Make AI test main compile