diff --git a/document_ocr/tests/test_document_ocr.py b/document_ocr/tests/test_document_ocr.py
index 8a72a9d9..393dbfa4 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 document_ocr.models.ir_attachment import _MARKER_PHRASE
+from ir_attachment import _MARKER_PHRASE
 from odoo.tests.common import TransactionCase