lib/ai: fix raw code in now tested docunit for puzzle test module