Added missing malloc include.