diff --git a/crates/plotnik-lib/src/query/mod.rs b/crates/plotnik-lib/src/query/mod.rs
index 3dccff31..6eb9976e 100644
--- a/crates/plotnik-lib/src/query/mod.rs
+++ b/crates/plotnik-lib/src/query/mod.rs
@@ -29,5 +29,3 @@ mod printer_tests;
 mod query_tests;
 #[cfg(test)]
 mod symbol_table_tests;
-#[cfg(test)]
-mod type_check_tests;
diff --git a/crates/plotnik-lib/src/query/type_check/mod.rs b/crates/plotnik-lib/src/query/type_check/mod.rs
index 74cc553b..9a5e116b 100644
--- a/crates/plotnik-lib/src/query/type_check/mod.rs
+++ b/crates/plotnik-lib/src/query/type_check/mod.rs
@@ -10,6 +10,9 @@ mod symbol;
 mod types;
 mod unify;
 
+#[cfg(test)]
+mod tests;
+
 pub use context::TypeContext;
 pub use emit_ts::{EmitConfig, TsEmitter, emit_typescript, emit_typescript_with_config};
 pub use symbol::{DefId, Interner, Symbol};
diff --git a/crates/plotnik-lib/src/query/type_check_tests.rs b/crates/plotnik-lib/src/query/type_check/tests.rs
similarity index 100%
rename from crates/plotnik-lib/src/query/type_check_tests.rs
rename to crates/plotnik-lib/src/query/type_check/tests.rs