diff --git a/tests/test_test_example.py b/tests/test_test_example.py
index ad15eee2..b707a6d9 100644
--- a/tests/test_test_example.py
+++ b/tests/test_test_example.py
@@ -1,4 +1,4 @@
-from zeta import MultiheadAttention
+
 
 import time
 import unittest