diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
index 58c0085a8a..0ebbf9f60c 100644
--- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
+++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
@@ -236,7 +236,7 @@ pub async fn omni_gen_image_generate_handler(
       for media_token in ref_tokens {
         context_items.push(PromptContextItem {
           media_token: media_token.clone(),
-          context_semantic_type: PromptContextSemanticType::Imgsrc,
+          context_semantic_type: PromptContextSemanticType::Imgref,
         });
       }
     }