diff --git a/server-rs/crates/api-server/src/openai_image_generation.rs b/server-rs/crates/api-server/src/openai_image_generation.rs
index ea026cf6..7f5a6707 100644
--- a/server-rs/crates/api-server/src/openai_image_generation.rs
+++ b/server-rs/crates/api-server/src/openai_image_generation.rs
@@ -2,13 +2,12 @@ use axum::http::StatusCode;
 use platform_image::{
     DownloadedImage, GeneratedImages, PlatformImageError, PlatformImageStatusHint, ReferenceImage,
     VECTOR_ENGINE_PROVIDER, VectorEngineImageSettings, build_vector_engine_image_http_client,
-    create_vector_engine_image_edit, create_vector_engine_image_edit_with_references,
-    create_vector_engine_image_generation,
+    build_vector_engine_image_request_body, create_vector_engine_image_edit,
+    create_vector_engine_image_edit_with_references, create_vector_engine_image_generation,
 };
 #[cfg(test)]
 use platform_image::{
-    build_vector_engine_image_request_body, vector_engine_images_edit_url,
-    vector_engine_images_generation_url,
+    vector_engine_images_edit_url, vector_engine_images_generation_url,
 };
 use serde_json::{Value, json};
 use time::OffsetDateTime;
@@ -236,7 +235,6 @@ pub(crate) async fn create_openai_image_edit_with_references(
     .await
 }
 
-#[cfg(test)]
 pub(crate) fn build_openai_image_request_body(
     prompt: &str,
     negative_prompt: Option<&str>,