diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp
index 8dbd350d253b9a..7c8dae85c448aa 100644
--- a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp
+++ b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.cpp
@@ -10,7 +10,6 @@
 
 #include <ngraph/opsets/opset2.hpp>
 
-#include <legacy/ie_layers.h>
 #include "ie_parallel.hpp"
 #include "utils/bfloat16.hpp"
 #include "emitters/jit_load_store_emitters.hpp"
@@ -19,7 +18,9 @@
 
 #include <string>
 #include <vector>
-#include <math.h>
+#include <memory>
+#include <algorithm>
+#include <cmath>
 
 using namespace MKLDNNPlugin;
 using namespace InferenceEngine;
@@ -375,7 +376,7 @@ void MKLDNNROIPoolingNode::initSupportedPrimitiveDescriptors() {
     if (!supportedPrimitiveDescriptors.empty())
         return;
 
-    runtimePrecision = getCnnLayer()->insData[0].lock()->getPrecision();
+    runtimePrecision = getOriginalInputPrecisionAtPort(0);
 
     if (!mayiuse(avx512_core)) {
         if (runtimePrecision == Precision::BF16)
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h
index 6a04dd4c2c2b79..7028e33e83de29 100644
--- a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h
+++ b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_roi_pooling_node.h
@@ -4,8 +4,9 @@
 
 #pragma once
 
-#include <ie_common.h>
 #include <mkldnn_node.h>
+#include <ie_common.h>
+
 #include <string>
 #include <memory>
 #include <vector>