Fix for malloc