diff --git a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_multiple_query_sdp.cpp b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_multiple_query_sdp.cpp
index d33eb9d81fb3cb..92366e6b85fb3a 100644
--- a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_multiple_query_sdp.cpp
+++ b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_multiple_query_sdp.cpp
@@ -269,7 +269,12 @@ class ConcatMultiQuerySDPTest : public testing::WithParamInterface<ConcatMultiQu
             outputs.push_back(copy);
         }
         auto states = inferRequest.query_state();
-        for (auto&& state : states) {
+        for (std::string name : {"pastk", "pastv"}) {
+            auto itr = std::find_if(states.begin(), states.end(), [&](const ov::VariableState& state) {
+                return name == state.get_name();
+            });
+            OPENVINO_ASSERT(itr != states.end(), "Failed to find ", name, " state");
+            const auto& state = *itr;
             auto state_tensor = state.get_state();
             ov::Tensor copy{state_tensor.get_element_type(), state_tensor.get_shape()};
             state_tensor.copy_to(copy);
diff --git a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_transpose_sdp_transpose.cpp b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_transpose_sdp_transpose.cpp
index 5158ff8f433d73..9bf5011404b175 100644
--- a/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_transpose_sdp_transpose.cpp
+++ b/src/plugins/intel_cpu/tests/functional/custom/subgraph_tests/src/concat_transpose_sdp_transpose.cpp
@@ -253,7 +253,12 @@ class ConcatSDPTransposeTest : public ConcatSDPTransposeTestBase {
             outputs.push_back(copy);
         }
         auto states = inferRequest.query_state();
-        for (auto&& state : states) {
+        for (std::string name : {"pastk", "pastv"}) {
+            auto itr = std::find_if(states.begin(), states.end(), [&](const ov::VariableState& state) {
+                return name == state.get_name();
+            });
+            OPENVINO_ASSERT(itr != states.end(), "Failed to find ", name, " state");
+            const auto& state = *itr;
             auto state_tensor = state.get_state();
             ov::Tensor copy{state_tensor.get_element_type(), state_tensor.get_shape()};
             state_tensor.copy_to(copy);