diff --git a/document_ocr/tests/test_document_ocr.py b/document_ocr/tests/test_document_ocr.py
index 393dbfa4..9765e291 100644
--- a/document_ocr/tests/test_document_ocr.py
+++ b/document_ocr/tests/test_document_ocr.py
@@ -5,7 +5,7 @@
 from StringIO import StringIO
 
 from PIL import Image, ImageDraw, ImageFont
-from ir_attachment import _MARKER_PHRASE
+from ..models.ir_attachment import _MARKER_PHRASE
 from odoo.tests.common import TransactionCase