Testing AI Image fixes