diff --git a/tests/test_embeddings.py b/tests/test_embeddings.py
index 31d705e..c9350c8 100644
--- a/tests/test_embeddings.py
+++ b/tests/test_embeddings.py
@@ -1,4 +1,3 @@
-import pytest
 import numpy as np
 
 from shell_history_search.core import EmbeddingService
@@ -87,4 +86,4 @@ class TestEmbeddingService:
         e1 = service.encode_single("same command")
 
         sim = EmbeddingService.cosine_similarity(e1, e1)
-        assert 0.9999 < sim <= 1.0001
\ No newline at end of file
+        assert 0.9999 < sim <= 1.0001