diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/mod.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/mod.rs
index e094c493df..a37bd531a2 100644
--- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/mod.rs
+++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/mod.rs
@@ -3,3 +3,6 @@ mod request_to_costs;
 mod request_to_plan;
 mod resolve_media_tokens;
 pub(crate) mod transform_request;
+
+#[cfg(test)]
+mod tests;
diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/tests/mod.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/tests/mod.rs
new file mode 100644
index 0000000000..e69de29bb2
diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/mod.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/mod.rs
index 212954a26e..38f5bfec13 100644
--- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/mod.rs
+++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/mod.rs
@@ -3,3 +3,6 @@ mod request_to_costs;
 mod request_to_plan;
 mod resolve_media_tokens;
 pub(crate) mod transform_request;
+
+#[cfg(test)]
+mod tests;
diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/tests/mod.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/tests/mod.rs
new file mode 100644
index 0000000000..e69de29bb2