diff --git a/tests/e2e/test_one_layer.py b/tests/e2e/test_one_layer.py
index 4ef2c253..3517b4f8 100644
--- a/tests/e2e/test_one_layer.py
+++ b/tests/e2e/test_one_layer.py
@@ -17,7 +17,6 @@ to when the compressor + indexer are wired (CSA layer test).
 """
 import torch
 import math
-import pytest
 
 
 def test_swa_cache_gather():