diff --git a/crates/stemedb-storage/src/similarity_index/tests.rs b/crates/stemedb-storage/src/similarity_index/tests.rs
index 70067d8..1675bef 100644
--- a/crates/stemedb-storage/src/similarity_index/tests.rs
+++ b/crates/stemedb-storage/src/similarity_index/tests.rs
@@ -1,7 +1,7 @@
 //! Tests for the SimilarityIndex implementation.
 
-use super::store_impl::*;
-use super::traits::SimilarityIndex;
+use super::*;
+use crate::similarity_index::traits::SimilarityIndex;
 use crate::HybridStore;
 use std::sync::Arc;