Yeah, that's a good idea. Right now the AI only takes a single hit, which often causes an anticlimactic win where the AI just gets hit by a stray bullet.
