Fix malloc error