diff --git a/testserver_o2b/models/init_test.py b/testserver_o2b/models/init_test.py
index 42e333e..f2fb2d6 100644
--- a/testserver_o2b/models/init_test.py
+++ b/testserver_o2b/models/init_test.py
@@ -8,6 +8,7 @@ import os
 import uuid
 import odoo
 import requests
+from odoo.exceptions import ValidationError
 
 
 class CustomModel(models.Model):