diff --git a/tricks/modules/ltx_model.py b/tricks/modules/ltx_model.py
index 59b62d9..05d337f 100644
--- a/tricks/modules/ltx_model.py
+++ b/tricks/modules/ltx_model.py
@@ -7,7 +7,6 @@
     BasicTransformerBlock,
     LTXVModel,
     apply_rotary_emb,
-    precompute_freqs_cis,
 )
 from comfy.ldm.lightricks.symmetric_patchifier import latent_to_pixel_coords
 from torch import nn
@@ -167,7 +166,7 @@ def forward(
                 (attention_mask.shape[0], 1, -1, attention_mask.shape[-1])
             ) * torch.finfo(x.dtype).max
 
-        pe = precompute_freqs_cis(
+        pe = self._precompute_freqs_cis(
             fractional_coords, dim=self.inner_dim, out_dtype=x.dtype
         )