Fixed malloc warning