diff --git a/matcha/models/components/text_encoder.py b/matcha/models/components/text_encoder.py
index 36fc65a..66e7164 100644
--- a/matcha/models/components/text_encoder.py
+++ b/matcha/models/components/text_encoder.py
@@ -6,7 +6,7 @@ import torch
 import torch.nn as nn  # pylint: disable=consider-using-from-import
 from einops import rearrange
 
-from matcha import utils
+import matcha.utils as utils  # pylint: disable=consider-using-from-import
 from matcha.utils.model import sequence_mask
 
 log = utils.get_pylogger(__name__)