Fixed malloc bugs