diff --git a/ivsr_sdk/src/smart_patch.cpp b/ivsr_sdk/src/smart_patch.cpp
index 5bc433e..b6a25f4 100644
--- a/ivsr_sdk/src/smart_patch.cpp
+++ b/ivsr_sdk/src/smart_patch.cpp
@@ -101,7 +101,7 @@ void fill_image(std::vector<std::vector<int>> patchCorners, char* imgBuf, \
     int img_sN = iC * iH * iW;
     int img_sB = iN * iC * iH * iW;
     
-    size_t outputpixels = iB * iN * iC * iH * iW;
+    size_t outputpixels = static_cast<size_t>(iB) * iN * iC * iH * iW;
     int * pixelCounter = new int[outputpixels]();
 
     float * img_ptr =(float *)imgBuf;