Fix AI, tweak stage0