diff --git a/projects/rocr-runtime/src/memory.c b/projects/rocr-runtime/src/memory.c
index ee60f6d775..55a67ff757 100644
--- a/projects/rocr-runtime/src/memory.c
+++ b/projects/rocr-runtime/src/memory.c
@@ -275,7 +275,7 @@ hsaKmtMapMemoryToGPUNodes(
 		return ret;
 
 	ret = fmm_register_memory(MemoryAddress, MemorySizeInBytes,
-			gpu_id_array, NumberOfNodes);
+			gpu_id_array, NumberOfNodes*sizeof(uint32_t));
 	if (ret != HSAKMT_STATUS_SUCCESS)
 		return ret;