diff --git a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
index 4156e0b..1357146 100644
--- a/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.cpp
@@ -25,7 +25,7 @@ upsampleLayer(int layerIdx, std::map<std::string, std::string>& block, nvinfer1:
   std::string resizeLayerName = "upsample_" + std::to_string(layerIdx);
   resize->setName(resizeLayerName.c_str());
   
-#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 0
+#if NV_TENSORRT_MAJOR >= 8 && NV_TENSORRT_MINOR > 4
   resize->setResizeMode(nvinfer1::InterpolationMode::kNEAREST);
 #else
   resize->setResizeMode(nvinfer1::ResizeMode::kNEAREST);