ai + test fixes