fix the malloc bug