diff --git a/device/device.cpp b/device/device.cpp
index 2709febd..8c86b86e 100644
--- a/device/device.cpp
+++ b/device/device.cpp
@@ -751,7 +751,7 @@ bool Device::disableP2P(amd::Device* ptrDev) {
 }
 
 bool Device::UpdateStackSize(uint64_t stackSize) {
-  if (stackSize > 16 * Ki) {
+  if (stackSize > ((128 * Ki) - 16)) {
     return false;
   }
   stack_size_ = stackSize;