diff --git a/src/main.rs b/src/main.rs
index 098772f..df77e89 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -72,6 +72,7 @@ mod playlist;
 mod serve;
 mod server;
 mod swagger;
+#[cfg(test)]
 mod test;
 mod thumbnails;
 mod ui;
diff --git a/src/swagger.rs b/src/swagger.rs
index 1d959c1..6a47a45 100644
--- a/src/swagger.rs
+++ b/src/swagger.rs
@@ -1,4 +1,3 @@
-use rocket::http::Status;
 use rocket::response::NamedFile;
 use rocket::State;
 use std::io;
@@ -6,7 +5,6 @@ use std::path::PathBuf;
 use std::sync::Arc;
 
 use crate::server::StaticDirs;
-use crate::test::get_test_environment;
 
 pub fn get_routes() -> Vec<rocket::Route> {
 	routes![
@@ -30,6 +28,9 @@ fn files(static_dirs: State<Arc<StaticDirs>>, file: PathBuf) -> Option<NamedFile
 
 #[test]
 fn test_index() {
+	use rocket::http::Status;
+	use crate::test::get_test_environment;
+
 	let env = get_test_environment("swagger_index.sqlite");
 	let client = &env.client;
 	let response = client.get("/swagger").dispatch();
diff --git a/src/web.rs b/src/web.rs
index 695b616..b935b71 100644
--- a/src/web.rs
+++ b/src/web.rs
@@ -1,4 +1,3 @@
-use rocket::http::Status;
 use rocket::response::NamedFile;
 use rocket::State;
 use std::io;
@@ -6,7 +5,6 @@ use std::path::PathBuf;
 use std::sync::Arc;
 
 use crate::server::StaticDirs;
-use crate::test::get_test_environment;
 
 pub fn get_routes() -> Vec<rocket::Route> {
 	routes![
@@ -30,6 +28,9 @@ fn files(static_dirs: State<Arc<StaticDirs>>, file: PathBuf) -> Option<NamedFile
 
 #[test]
 fn test_index() {
+	use rocket::http::Status;
+	use crate::test::get_test_environment;
+
 	let env = get_test_environment("web_index.sqlite");
 	let client = &env.client;
 	let response = client.get("/").dispatch();