diff --git a/hipamd/src/hip_device_runtime.cpp b/hipamd/src/hip_device_runtime.cpp
index c453820f66..d6561b8277 100644
--- a/hipamd/src/hip_device_runtime.cpp
+++ b/hipamd/src/hip_device_runtime.cpp
@@ -514,9 +514,7 @@ hipError_t hipDeviceGetLimit(size_t* pValue, hipLimit_t limit) {
 
   switch (limit) {
     case hipLimitMallocHeapSize:
-      hipDeviceProp_tR0600 prop;
-      HIP_RETURN_ONFAIL(ihipGetDeviceProperties(&prop, ihipGetDevice()));
-      *pValue = prop.totalGlobalMem;
+      *pValue = hip::getCurrentDevice()->devices()[0]->InitialHeapSize();
       break;
     case hipLimitStackSize:
       *pValue = hip::getCurrentDevice()->devices()[0]->StackSize();