diff --git a/samples/ObjectDetection/include/YoloResultDecoder.hpp b/samples/ObjectDetection/include/YoloResultDecoder.hpp
index 98435e3..ae6cb5e 100644
--- a/samples/ObjectDetection/include/YoloResultDecoder.hpp
+++ b/samples/ObjectDetection/include/YoloResultDecoder.hpp
@@ -26,9 +26,9 @@
      */
     YoloResultDecoder(float NMSThreshold, float ClsThreshold, float ObjectThreshold);
 
-    DetectedObjects Decode(const InferenceResults& results,
-                           const Size& outputFrameSize,
-                           const Size& resizedFrameSize,
+    DetectedObjects Decode(const common::InferenceResults<float>& results,
+                           const common::Size& outputFrameSize,
+                           const common::Size& resizedFrameSize,
                            const std::vector <std::string>& labels) override;
 private:
     float m_NmsThreshold;