diff --git a/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_offsets_sum.cpp b/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_offsets_sum.cpp
index ef1542e2a2bd23..acbcda3a108286 100644
--- a/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_offsets_sum.cpp
+++ b/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_offsets_sum.cpp
@@ -3,7 +3,7 @@
 //
 
 #include "shared_test_classes/single_op/embedding_bag_offsets_sum.hpp"
-#include "ngraph_functions/builders.hpp"
+#include "ov_models/builders.hpp"
 
 namespace ov {
 namespace test {
diff --git a/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_packed_sum.cpp b/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_packed_sum.cpp
index 788271a4f4aefc..ab849f0f7d04f7 100644
--- a/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_packed_sum.cpp
+++ b/src/tests/functional/shared_test_classes/src/single_op/embedding_bag_packed_sum.cpp
@@ -3,7 +3,7 @@
 //
 
 #include "shared_test_classes/single_op/embedding_bag_packed_sum.hpp"
-#include "ngraph_functions/builders.hpp"
+#include "ov_models/builders.hpp"
 
 namespace ov {
 namespace test {
diff --git a/src/tests/functional/shared_test_classes/src/single_op/embedding_segments_sum.cpp b/src/tests/functional/shared_test_classes/src/single_op/embedding_segments_sum.cpp
index 71a80bd7082514..b3ded1f3d9af36 100644
--- a/src/tests/functional/shared_test_classes/src/single_op/embedding_segments_sum.cpp
+++ b/src/tests/functional/shared_test_classes/src/single_op/embedding_segments_sum.cpp
@@ -3,7 +3,7 @@
 //
 
 #include "shared_test_classes/single_op/embedding_segments_sum.hpp"
-#include "ngraph_functions/builders.hpp"
+#include "ov_models/builders.hpp"
 
 
 namespace ov {