diff --git a/tests/conftest.py b/tests/conftest.py
index 98b192e..7dd3d4e 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -20,9 +20,10 @@ def sample_history_entry():
 def sample_search_result():
     return SearchResult(
         command="ls -la",
-        timestamp=1234567890.0,
-        score=0.95,
-        shell="bash"
+        shell_type="bash",
+        timestamp=1234567890,
+        similarity=0.95,
+        command_id=1
     )