Fix malloc bug