diff --git a/tests/test_x_transformers.py b/tests/test_x_transformers.py
index 7cf91f4..85d47e8 100644
--- a/tests/test_x_transformers.py
+++ b/tests/test_x_transformers.py
@@ -1,5 +1,3 @@
-from contextlib import nullcontext
-
 import pytest
 import torch