Add malloc cast