Malloc size fix