diff --git a/vall_e/models/arch/attention/fused.py b/vall_e/models/arch/attention/fused.py
index 633cf3e..34de5a7 100644
--- a/vall_e/models/arch/attention/fused.py
+++ b/vall_e/models/arch/attention/fused.py
@@ -14,7 +14,6 @@ Extra Credits:
 
 """
 
-import pytest
 import torch
 
 import triton