Sure. You can always write more tests. That's not a problem specific to AI.

I'd also do code reviews on the code AI produces.