Fix malloc type