diff --git a/tests/models/udop/test_processor_udop.py b/tests/models/udop/test_processor_udop.py
index 05855991b18..5e913f8ec6d 100644
--- a/tests/models/udop/test_processor_udop.py
+++ b/tests/models/udop/test_processor_udop.py
@@ -22,7 +22,6 @@ from typing import List
 import numpy as np
 
 from transformers import PreTrainedTokenizer, PreTrainedTokenizerBase, PreTrainedTokenizerFast
-from transformers.models.udop import UdopTokenizer, UdopTokenizerFast
 from transformers.testing_utils import (
     require_pytesseract,
     require_sentencepiece,
@@ -35,11 +34,11 @@ from transformers.utils import FEATURE_EXTRACTOR_NAME, cached_property, is_pytes
 
 if is_torch_available():
     import torch
+    from transformers import UdopTokenizer, UdopTokenizerFast
 
 
 if is_pytesseract_available():
     from PIL import Image
-
     from transformers import LayoutLMv3ImageProcessor, UdopProcessor