Mem alloc size fix