From 136bb81f5045debf7fcf9f06f50fae2c91a84038 Mon Sep 17 00:00:00 2001
From: Varun Jain <varunudr@amazon.com>
Date: Tue, 2 Jan 2024 02:11:24 -0800
Subject: [PATCH] Bug fix of load model id

Signed-off-by: Varun Jain <varunudr@amazon.com>
---
 .../java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
index b9c2c67da..00ff46733 100644
--- a/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
+++ b/qa/rolling-upgrade/src/test/java/org/opensearch/neuralsearch/bwc/SemanticSearchIT.java
@@ -54,6 +54,7 @@ public void testSemanticSearch_E2EFlow() throws Exception {
             case UPGRADED:
                 modelId = getModelId(PIPELINE_NAME);
                 int totalDocsCountUpgraded = 3 * NUM_DOCS_PER_ROUND;
+                loadModel(modelId);
                 addDocument(getIndexNameForTest(), "2", TEST_FIELD, TEXT_UPGRADED);
                 validateTestIndexOnUpgrade(totalDocsCountUpgraded, modelId, TEXT_UPGRADED);
                 deletePipeline(PIPELINE_NAME);