AI test tweak